numpy.expand_dims

numpy.expand_dims(a, axis)

展开数组的形状.

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

参数:

aarray_like

输入数组.

axisint 或 ints 元组

新轴所在的展开轴中的位置.

自 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

示例

Try it in your browser!

>>> 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

Go BackOpen In Tab