python 减少numpy维度

要减少NumPy数组的维度，可以使用squeeze函数。squeeze函数的作用是删除数组中的单维度条目，即将形状中大小为1的维度去除。以此可以降低数组的维度。

以下是一个示例代码：

import numpy as np

# 创建一个具有单维度的NumPy数组
x = np.array([[[1], [2], [3]], [[4], [5], [6]]])
print("初始数组的形状:", x.shape)  # 输出 (2, 3, 1)

# 使用squeeze函数减少数组的维度
y = np.squeeze(x)
print("减少维度后的数组形状:", y.shape)  # 输出 (2, 3)

在上述示例中，我们创建了一个形状为 (2, 3, 1) 的NumPy数组，并使用squeeze函数将其维度减少为 (2, 3)。通过这个函数，我们可以方便地降低NumPy数组的维度。

参考资料：
NumPy squeeze文档：https://numpy.org/doc/stable/reference/generated/numpy.squeeze.html?highlight=squeeze#numpy.squeeze