可以使用 numpy 的 squeeze()
函数去掉维度为 1 的维度。使用方法如下:
import numpy as npa = np.array([[[1], [2], [3]], [[4], [5], [6]]])
b = np.squeeze(a)
print(b)
输出结果:
[[1 2 3]
[4 5 6]]
可以看到原始数组 a
的第一维和第三维都是维度为 1 的维度,在使用 squeeze()
函数后被去掉了。
需要注意的是,如果使用 np.squeeze(a,axis=...)
参数可以指定去除特定维度。