共計 443 個字符，預計需要花費 2 分鐘才能閱讀完成。

在 Python 中，squeeze()函數(shù)用于去除數(shù)組中的單維度條目。具體來說，它會刪除 shape 為 1 的維度，即將維度大小為 1 的維度去掉。

例如，假設(shè)有一個形狀為 (1, 5) 的數(shù)組，使用 squeeze() 函數(shù)后，它將變成形狀為 (5,) 的數(shù)組。

import numpy as np

arr = np.array([[1, 2, 3, 4, 5]])
print(arr.shape)  # 輸出 (1, 5)

arr_squeezed = np.squeeze(arr)
print(arr_squeezed.shape)  # 輸出 (5,)

squeeze()函數(shù)還可以指定去除某個特定的維度，例如 axis=0 表示去除第一個維度。

import numpy as np

arr = np.array([[[1, 2, 3, 4, 5]]])
print(arr.shape)  # 輸出 (1, 1, 5)

arr_squeezed = np.squeeze(arr, axis=0)
print(arr_squeezed.shape)  # 輸出 (1, 5)

丸趣 TV 網(wǎng) – 提供最優(yōu)質(zhì)的資源集合！