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()
标签：维度,plt,print,数组,np,squeeze From： https://www.cnblogs.com/keye/p/16619241.html