python中squeeze函数的用法是什么

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