numpy 如何使用np.newaxis?

2w2cym1i  于 2023-10-19  发布在  其他
关注(0)|答案(5)|浏览(107)

什么是numpy.newaxis,什么时候应该使用它?
在一维数组x上使用它会产生:

>>> x
array([0, 1, 2, 3])

>>> x[np.newaxis, :]
array([[0, 1, 2, 3]])

>>> x[:, np.newaxis]
array([[0],
       [1],
       [2],
       [3]])
sbtkgmzw

sbtkgmzw1#

简单地说,numpy.newaxis用于***将现有数组的维度***增加 * 一个维度 *,使用 * 一次 *。因此,在本发明中,

*1D阵列将变为2D阵列
*2D阵列将变为3D阵列
*3D阵列将变为4D阵列
*4D阵列将变为5D阵列

等等。
下面是一个直观的插图,描述了1D数组到2D数组的 * 升级 *。

场景一np.newaxis可能会派上用场,当你想 * 显式 * 转换一个一维数组为 * 行向量 * 或 * 列向量 *,如上图所示。
示例:

# 1D array
In [7]: arr = np.arange(4)
In [8]: arr.shape
Out[8]: (4,)

# make it as row vector by inserting an axis along first dimension
In [9]: row_vec = arr[np.newaxis, :]     # arr[None, :]
In [10]: row_vec.shape
Out[10]: (1, 4)

# make it as column vector by inserting an axis along second dimension
In [11]: col_vec = arr[:, np.newaxis]     # arr[:, None]
In [12]: col_vec.shape
Out[12]: (4, 1)

场景2:当我们想使用numpy broadcasting作为某些操作的一部分时,例如在做一些数组的 * 加法 * 时。
示例:

假设你想添加以下两个数组:

x1 = np.array([1, 2, 3, 4, 5])
 x2 = np.array([5, 4, 3])

如果您尝试像这样添加这些,NumPy将引发以下ValueError

ValueError: operands could not be broadcast together with shapes (5,) (3,)

在这种情况下,您可以使用np.newaxis来增加其中一个数组的维度,以便NumPy可以broadcast

In [2]: x1_new = x1[:, np.newaxis]    # x1[:, None]
# now, the shape of x1_new is (5, 1)
# array([[1],
#        [2],
#        [3],
#        [4],
#        [5]])

现在,添加:

In [3]: x1_new + x2
Out[3]:
array([[ 6,  5,  4],
       [ 7,  6,  5],
       [ 8,  7,  6],
       [ 9,  8,  7],
       [10,  9,  8]])

或者,您也可以向数组x2添加新轴:

In [6]: x2_new = x2[:, np.newaxis]    # x2[:, None]
In [7]: x2_new     # shape is (3, 1)
Out[7]: 
array([[5],
       [4],
       [3]])

现在,添加:

In [8]: x1 + x2_new
Out[8]: 
array([[ 6,  7,  8,  9, 10],
       [ 5,  6,  7,  8,  9],
       [ 4,  5,  6,  7,  8]])

注意:注意我们在两种情况下得到相同的结果(但一个是另一个的转置)。
场景3:这类似于场景1。但是,您可以多次使用np.newaxis来将数组提升到更高的维度。这种操作有时需要更高阶的数组(* 即,Tensor *)。
示例:

In [124]: arr = np.arange(5*5).reshape(5,5)

In [125]: arr.shape
Out[125]: (5, 5)

# promoting 2D array to a 5D array
In [126]: arr_5D = arr[np.newaxis, ..., np.newaxis, np.newaxis]    # arr[None, ..., None, None]

In [127]: arr_5D.shape
Out[127]: (1, 5, 5, 1, 1)

作为替代方案,您可以使用具有直观axis kwarg的numpy.expand_dims

# adding new axes at 1st, 4th, and last dimension of the resulting array
In [131]: newaxes = (0, 3, -1)
In [132]: arr_5D = np.expand_dims(arr, axis=newaxes)
In [133]: arr_5D.shape
Out[133]: (1, 5, 5, 1, 1)

更多关于np.newaxis与np.reshape的背景

newaxis也被称为伪索引,允许将轴临时添加到多数组中。
np.newaxis使用slicing操作符重新创建数组,而numpy.reshape将数组整形为所需的布局(假设尺寸匹配;这是 * 必须 * 的reshape发生)。

示例

In [13]: A = np.ones((3,4,5,6))
In [14]: B = np.ones((4,6))
In [15]: (A + B[:, np.newaxis, :]).shape     # B[:, None, :]
Out[15]: (3, 4, 5, 6)

在上面的例子中,我们在B的第一个和第二个轴之间插入了一个临时轴(以使用广播)。这里使用np.newaxis填充缺少的轴,以使broadcasting操作工作。

  • 一般提示 *:您也可以使用None代替np.newaxis;实际上是same objects
In [13]: np.newaxis is None
Out[13]: True

P.S.也可以看到这个很棒的答案:newaxis vs reshape to add dimensions

am46iovg

am46iovg2#

什么是np.newaxis

np.newaxis只是Python常量None的别名,这意味着无论你在哪里使用np.newaxis,你也可以使用None

>>> np.newaxis is None
True

如果你读到的代码使用的是np.newaxis而不是None,那么它就更有描述性了。

如何使用np.newaxis

np.newaxis通常用于切片。它指示您要向数组添加额外的维度。np.newaxis的位置表示我想添加维度的位置。

>>> import numpy as np
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> a.shape
(10,)

在第一个例子中,我使用了第一维中的所有元素,并添加了第二维:

>>> a[:, np.newaxis]
array([[0],
       [1],
       [2],
       [3],
       [4],
       [5],
       [6],
       [7],
       [8],
       [9]])
>>> a[:, np.newaxis].shape
(10, 1)

第二个示例添加一个维度作为第一个维度,然后使用原始数组第一个维度中的所有元素作为结果数组第二维度中的元素:

>>> a[np.newaxis, :]  # The output has 2 [] pairs!
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
>>> a[np.newaxis, :].shape
(1, 10)

同样,您可以使用多个np.newaxis来添加多个维:

>>> a[np.newaxis, :, np.newaxis]  # note the 3 [] pairs in the output
array([[[0],
        [1],
        [2],
        [3],
        [4],
        [5],
        [6],
        [7],
        [8],
        [9]]])
>>> a[np.newaxis, :, np.newaxis].shape
(1, 10, 1)

np.newaxis的替代品吗?

NumPy中还有一个非常相似的功能:np.expand_dims,也可用于插入一维:

>>> np.expand_dims(a, 1)  # like a[:, np.newaxis]
>>> np.expand_dims(a, 0)  # like a[np.newaxis, :]

但是考虑到它只是在shape中插入1 s,你也可以reshape数组来添加这些维度:

>>> a.reshape(a.shape + (1,))  # like a[:, np.newaxis]
>>> a.reshape((1,) + a.shape)  # like a[np.newaxis, :]

大多数情况下,np.newaxis是添加维度的最简单方法,但最好知道替代方法。

什么时候使用np.newaxis

在几种情况下,添加尺寸标注是有用的:

  • 如果数据应该具有指定的维数。例如,如果你想使用matplotlib.pyplot.imshow来显示一个1D数组。
  • 如果你想让NumPy广播数组。例如,通过添加维度,你可以得到一个数组中所有元素之间的差异:a - a[:, np.newaxis] .这是因为NumPy操作从最后一个维度1开始广播。
  • 添加必要的维度,以便NumPy * 可以 * 广播数组。这样做是因为每个长度为1的维度只是简单地广播到另一个数组的相应1维度的长度。

1如果你想了解更多关于广播规则的信息,NumPy关于这个主题的文档非常好。它还包括一个np.newaxis的例子:

>>> a = np.array([0.0, 10.0, 20.0, 30.0])
>>> b = np.array([1.0, 2.0, 3.0])
>>> a[:, np.newaxis] + b
array([[  1.,   2.,   3.],
       [ 11.,  12.,  13.],
       [ 21.,  22.,  23.],
       [ 31.,  32.,  33.]])
xzabzqsa

xzabzqsa3#

你从一维的数字列表开始。一旦使用了numpy.newaxis,就可以将其转换为一个二维矩阵,由四行一列组成。
然后,您可以使用该矩阵进行矩阵乘法,或者将其用于构建更大的4 x n矩阵。

htzpubme

htzpubme4#

选择元组中的newaxis对象用于将结果选择的维度扩展一个单位长度**维度。
它不仅仅是将行矩阵转换为列矩阵。
考虑下面的示例:

In [1]:x1 = np.arange(1,10).reshape(3,3)
       print(x1)
Out[1]: array([[1, 2, 3],
               [4, 5, 6],
               [7, 8, 9]])

现在让我们为我们的数据添加新的维度,

In [2]:x1_new = x1[:,np.newaxis]
       print(x1_new)
Out[2]:array([[[1, 2, 3]],

              [[4, 5, 6]],

              [[7, 8, 9]]])

你可以看到newaxis在这里添加了额外的维度,x1的维度是(3,3),X1_new的维度是(3,1,3)。
我们的新维度如何使我们能够进行不同的操作:

In [3]:x2 = np.arange(11,20).reshape(3,3)
       print(x2)
Out[3]:array([[11, 12, 13],
              [14, 15, 16],
              [17, 18, 19]])

加上x1_new和x2,我们得到:

In [4]:x1_new+x2
Out[4]:array([[[12, 14, 16],
               [15, 17, 19],
               [18, 20, 22]],

              [[15, 17, 19],
               [18, 20, 22],
               [21, 23, 25]],

              [[18, 20, 22],
               [21, 23, 25],
               [24, 26, 28]]])

因此,newaxis不仅仅是行到列矩阵的转换。它增加了矩阵的维数,从而使我们能够对它进行更多的操作。

qzwqbdag

qzwqbdag5#

它会在它所在的地方变暗。
举例来说:arr[newaxis]等于expand_dims(arr, 0)
arr[:, newaxis]等于expand_dims(arr, 1)
arr[:, :, newaxis]等于expand_dims(arr, 2)

相关问题