numpy.newaxis 方法

Muhammad Maisam Abbas 2021年7月4日
numpy.newaxis 方法

本教程将讨论 numpy.newaxis 方法。

numpy.newaxis 方法

numpy.newaxis 方法是 None 的别名,用于 Python 中的数组索引。numpy.newaxis 最直接的用法是在 Python 中为 NumPy 数组添加一个新维度。例如,将一维数组转换为二维数组,将二维数组转换为 3D 数组,等等。我们还可以使用 Python 中的 numpy.newaxis 将一维数组转换为行矩阵或列矩阵。以下代码示例向我们展示了如何在 Python 中使用 numpy.newaxis 将一维数组转换为行矩阵。

import numpy as np

array = np.array([1,2,3,4])
print(array.shape)

array = array[np.newaxis]
print(array.shape)

输出:

(4,)
(1, 4)

在上面的代码中,我们将一维数组 array 转换为带有 np.newaxis 的行矩阵。我们首先使用 np.array() 函数创建了一维数组 array。然后我们使用 [np.newaxis] 作为 array 的索引来返回一个行矩阵。以下代码示例向我们展示了如何使用 numpy.newaxis 将一维数组转换为列矩阵。

import numpy as np

array = np.array([1,2,3,4])
print(array.shape)

array = array[:,np.newaxis]
print(array.shape)

输出:

(4,)
(4, 1)

在上面的代码中,我们将一维数组 array 转换为带有 np.newaxis 的列矩阵。我们首先使用 np.array() 函数创建了一维数组 array。然后我们使用 [:, np.newaxis] 作为 array 的索引来返回一个列矩阵。

正如我们之前提到的,numpy.newaxisNone 的别名。因此,我们也可以使用 None 代替 numpy.newaxis 来执行上面讨论的两种操作。下面的代码示例向我们展示了如何在 Python 中将一维数组转换为带有 None 的行矩阵。

import numpy as np

array = np.array([1,2,3,4])
print(array.shape)

array = array[None]
print(array.shape)

输出:

(4,)
(1, 4)

在上面的代码中,我们将一维数组 array 转换为带有 None 的行矩阵。我们首先使用 np.array() 函数创建了一维数组 array。然后我们使用 [None] 作为 array 的索引来返回一个行矩阵。

下面的代码示例向我们展示了如何在 Python 中将一维数组转换为带有 None 的列矩阵。

import numpy as np

array = np.array([1,2,3,4])
print(array.shape)

array = array[:, None]
print(array.shape)

输出:

(4,)
(4, 1)

在上面的代码中,我们将一维数组 array 转换为带有 None 的列矩阵。我们首先使用 np.array() 函数创建了一维数组 array。然后我们使用 [:, None] 作为 array 的索引来返回一个列矩阵。

总之,numpy.newaxisNone 是相同的,可用于数组索引和切片。

Muhammad Maisam Abbas avatar Muhammad Maisam Abbas avatar

Maisam is a highly skilled and motivated Data Scientist. He has over 4 years of experience with Python programming language. He loves solving complex problems and sharing his results on the internet.

LinkedIn