在Python中,squeeze()
函數用于去除數組中的單維度條目。具體來說,它會刪除shape為1的維度,即將維度大小為1的維度去掉。
例如,假設有一個形狀為(1, 5)的數組,使用squeeze()
函數后,它將變成形狀為(5,)的數組。
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()
函數還可以指定去除某個特定的維度,例如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)