上一主题

numpy.ma.atleast_3d

下一主题

numpy.ma.squeeze

numpy.ma.expand_dims

numpy.ma.expand_dims(x, axis)[source]

展开数组的形状。

通过在由axis参数指定的轴之前包括一个新轴,来扩展数组的形状。此函数的行为与numpy.expand_dims相同,但保留了屏蔽的元素。

也可以看看

numpy.expand_dims
顶级NumPy模块中的等效函数。

例子

>>> import numpy.ma as ma
>>> x = ma.array([1, 2, 4])
>>> x[1] = ma.masked
>>> x
masked_array(data = [1 -- 4],
             mask = [False  True False],
       fill_value = 999999)
>>> np.expand_dims(x, axis=0)
array([[1, 2, 4]])
>>> ma.expand_dims(x, axis=0)
masked_array(data =
 [[1 -- 4]],
             mask =
 [[False  True False]],
       fill_value = 999999)

使用np.newaxis的分片语法可以实现相同的结果。

>>> x[np.newaxis, :]
masked_array(data =
 [[1 -- 4]],
             mask =
 [[False  True False]],
       fill_value = 999999)