np.squeeze

np.squeeze：从数组的形状中删除维度为 $1$ 的维度。

np.squeeze(arr, axis)

参数：

• arr：输入数组
• axis：整数或整数元组，用于选择形状中一维维度的子集。

示例：

import numpy as np

x = np.arange(9).reshape(1, 3, 3)
print(x)

y = np.squeeze(x)
print(y)

print(x.shape, y.shape)

[[[0 1 2]
  [3 4 5]
  [6 7 8]]]
[[0 1 2]
 [3 4 5]
 [6 7 8]]
(1, 3, 3) (3, 3)

在用 matplotlib 直接画图，会报错，可以利用squeeze()函数将表示向量的数组转换为秩为1的数组。

plt.plot(x)
plt.show()

ValueError: x and y can be no greater than 2-D, but have shapes (1,) and (1, 3, 3)

plt.plot(y)
plt.show()