在Python中，squeeze()函數(shù)用于去除數(shù)組中的單維度條目。具體來說，它會刪除shape為1的維度，即將維度大小為1的維度去掉。
例如，假設有一個形狀為(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)
億速云公眾號
手機網(wǎng)站二維碼