NumPy .squeeze() 命令：删除单维度条目

NumPy 的 .squeeze() 命令的作用是从数组的形状中删除单维度条目，即将维度为 1 的维度去掉。例如，对于形状为 (1, 3, 1) 的数组，使用 squeeze() 命令将返回形状为 (3,) 的数组。

使用场景

• 当您需要将多维数组转换为一维数组时。
• 当您需要移除数组中不必要的维度时。

示例

import numpy as np

arr = np.array([[[1, 2, 3]]])
print('原始数组的形状：', arr.shape)  # 输出： (1, 1, 3)

squeezed_arr = np.squeeze(arr)
print('压缩后的数组的形状：', squeezed_arr.shape)  # 输出： (3,)

注意

• 如果数组有多个维度为 1，则 .squeeze() 命令将删除所有这些维度。
• 如果数组没有维度为 1，则 .squeeze() 命令将不改变数组的形状。