Python numpy squeeze用法及代碼示例

本文簡要介紹 python 語言中 numpy.squeeze 的用法。

用法:

numpy.squeeze(a, axis=None)

從 a 中刪除長度為 1 的軸。

參數：

a： array_like

輸入數據。

axis： 無或int 或整數元組，可選

選擇形狀中長度為 1 的條目的子集。如果選擇了形狀條目大於 1 的軸，則會引發錯誤。

返回：

squeezed： ndarray

輸入數組，但刪除了長度為 1 的所有維度或維度的子集。這始終是一個本身或一個視圖成一個。請注意，如果所有軸都被擠壓，則結果是 0d 數組而不是標量。

拋出：

ValueError

如果軸不是無，並且被擠壓的軸的長度不是 1

例子：

>>> 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