numpy 去掉 维度是1 的维度、

可以使用 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=...) 参数可以指定去除特定维度。

你可能感兴趣的:(numpy,python,机器学习,深度学习,数据分析)