numpy.expand_dims#

numpy.expand_dims(a, axis)[源代码]#

扩展数组的形状。

在扩展后的数组形状的 axis 位置插入一个新轴。

参数:
a类数组对象

输入数组。

axisint 或 int 的元组

将新轴（或多个轴）放置在扩展轴中的位置。

自 1.13.0 版本起已弃用： 传递 axis > a.ndim 的轴将被视为 axis == a.ndim，传递 axis < -a.ndim - 1 的轴将被视为 axis == 0。此行为已被弃用。

返回:
resultndarray

维度数量增加的 a 的视图。

另请参阅

squeeze

逆操作，移除单例维度

reshape

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

atleast_1d, atleast_2d, atleast_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
在此页面