numpy.expand_dims #

麻木的。Expand_dims ( a , axis ) [来源] #

扩展数组的形状。

插入一个新轴,该轴将出现在展开的数组形状中的轴位置处。

参数
类似数组

输入数组。

int 或 int 元组

在扩展轴中放置新轴(或多个轴)的位置。

自版本 1.13.0 起已弃用:传递轴 where将被视为 ,并且传递将被视为。此行为已被弃用。axis > a.ndimaxis == a.ndimaxis < -a.ndim - 1axis == 0

版本 1.18.0 中进行了更改:现在支持轴元组。现在禁止如上所述的超出范围的轴并引发AxisError.

返回
结果数组

维数增加的a视图。

也可以看看

squeeze

逆操作,删除单一维度

reshape

插入、删除和组合尺寸,以及调整现有尺寸的大小

doc.indexing, atleast_1d, atleast_2d,atleast_3d

例子

>>> x = np.array([1, 2])
>>> x.shape
(2,)

以下等价于或:x[np.newaxis, :]x[np.newaxis]

>>> y = np.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)

以下相当于:x[:, np.newaxis]

>>> y = np.expand_dims(x, axis=1)
>>> y
array([[1],
       [2]])
>>> y.shape
(2, 1)

axis也可能是一个元组:

>>> y = np.expand_dims(x, axis=(0, 1))
>>> y
array([[[1, 2]]])
>>> y = np.expand_dims(x, axis=(2, 0))
>>> y
array([[[1],
        [2]]])

请注意,某些示例可能会使用None来代替np.newaxis.这些是相同的对象:

>>> np.newaxis is None
True