numpy.ma.expand_dims#

ma.expand_dims(a, axis)[source]#

扩展数组的形状。

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

参数:
aarray_like

输入数组。

axisint 或 int 元组

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

版本 1.13.0 中已弃用: 传递一个 axis > a.ndim 的轴将被视为 axis == a.ndim,传递 axis < -a.ndim - 1 的轴将被视为 axis == 0。此行为已弃用。

版本 1.18.0 中的更改: 现在支持轴的元组。如上所述,超出范围的轴现在是被禁止的,并会引发 AxisError

返回值:
resultndarray

维度增加的 a 的视图。

参见

squeeze

逆运算,删除单一维度

reshape

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

atleast_1datleast_2datleast_3d

示例

>>> import numpy as np
>>> 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