numpy.squeeze

numpy.squeeze(a, axis=None)

从数组的形状中删除单维条目。

参数：

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)