numpy.squeeze
numpy.squeeze(a, axis=None) [source]
从数组shape中删除一维条目。
参数 : | a :array_like 输入数据。 axis : 1.7.0版中的新功能。 选择形状中一维条目的子集。 如果选择的形状输入大于一个的轴,则会引发错误。 |
返回值 : | squeezed :ndarray 输入数组,但删除了长度为1的全部或部分维度。 可能始终是其本身或a的视图。 请注意,如果所有轴都受到squeeze,则结果为0d数组,而不是标量。 |
Raises : | ValueError 如果axis不为 |
例子
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=0).shape
(3, 1)
>>> np.squeeze(x, axis=1).shape
Traceback (most recent call last):
...
ValueError: cannot select an axis to squeeze out which has size not equal to one
>>> np.squeeze(x, axis=2).shape
(1, 3)
>>> x = np.array([[1234]])
>>> x.shape
(1, 1)
>>> np.squeeze(x)
array(1234) # 0d array
>>> np.squeeze(x).shape
()
>>> np.squeeze(x)[()]
1234