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)