numpy.squeeze

原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.squeeze.html

译者:飞龙 UsyiyiCN

校对:(虚位以待)

numpy.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)