numpy数组重塑问题

数据挖掘 麻木的 图像预处理 重塑
2022-02-28 20:25:58

假设我有形状的图像数据(32,32,3)50000

  1. 如果我想将其重塑为(50000,3,32,32)我该怎么办?

我试过np.transpose(0, 3, 1, 2)但失败了。

  1. 如果我想打印号码3(50000,3,32,32)我该怎么办?
1个回答

如果image是一个 numpy 对象:

image = image.reshape((50000,3,32,32))

然后打印:

print(image[:,3,:,:])