python中squeeze函数的用法是什么

共计 443 个字符，预计需要花费 2 分钟才能阅读完成。

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

丸趣 TV 网 – 提供最优质的资源集合！