numpy.ma.squeeze¶
-
numpy.ma.
squeeze
(a, axis=None)[source]¶ 从数组的形状中删除单维条目。
参数: a:array_like
输入数据。
axis:无或int或tuple ints,可选
版本1.7.0中的新功能。
选择形状中单维条目的子集。如果选择形状输入大于1的轴,则会出现错误。
返回: 挤压:ndarray
输入数组,但删除了长度为1的所有或维度的子集。这总是a本身或到a的视图。
例子
>>> x = np.array([[[0], [1], [2]]]) >>> x.shape (1, 3, 1) >>> np.squeeze(x).shape (3,) >>> np.squeeze(x, axis=(2,)).shape (1, 3)