python - 在 numpy 中 reshape 数组

Question

我有一个大小为 5000x32x32x3 的 numpy 数组。数字 5000 是图片的数量，每张图片的宽度和高度都是 32x32，并且有 3 颜色 channel 。

现在我想以一种保留数据的方式创建一个形状为 5000x3x32x32 的 numpy 数组。我所说的保留数据的意思是:

1. 结果数组中应该有 5000 个数据点
2. 数组的第 2 维 (3) 正确确定了颜色 channel ，即第 2 维为 0 的所有元素都属于红色 channel ，其第 2 维为 1属于绿色 channel ，其2维2属于蓝色 channel 。

简单地通过 np.reshape(data,(5000,3,32,32)) reshape 是行不通的，因为它不会保留 channel ，而只是将数据 reshape 为所需的形状。

Answer 1

我认为您正在寻找轴的排列，numpy.transpose 可以完成这项工作:

data = np.transpose( data, (0, 3, 1, 2))