numpy.expand_dims

numpy.expand_dims(a, axis)

扩展数组的形状。

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

参数：

aarray_like

输入数组。

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