Python NumPy squeeze方法用法及代碼示例

例子

基本用法

假設我們有以下二維數組：

a = np.array([[4],[5],[6]])
a



array([[4],
       [5],
       [6]])

您能看出在這種情況下如何將 a 簡單地表示為一維數組 [4,5,6] 嗎？ squeeze(~) 方法允許我們執行此操作：

a.squeeze()



array([4, 5, 6])

同樣的原則也適用於以下情況：

a = np.array([[[1,2],[3,4]]])
a



array([[[1, 2],
        [3, 4]]])

在這裏，我們有一個 3D 數組，但很明顯它可以簡化為 2D 數組，如下所示：

a.squeeze()



array([[1, 2],
       [3, 4]])

僅壓縮軸的子集

假設我們有以下 3D 數組：

a = np.array([[[4],[5],[6]]])
a



array([[[4],
        [5],
        [6]]])

很明顯，這可以簡化為二維數組，甚至可以簡化為一維數組。我們可以使用 axis 參數指定要減少到哪個維度。

簡化為二維數組：

np.squeeze(a, axis=0)



array([[4],
       [5],
       [6]])

請注意，默認情況下，所有冗餘軸都會被刪除：

np.squeeze(a)



array([4, 5, 6])