python - 如何重新排列numpy中的值顺序？

Question

我有一个包含许多图像的文件夹(按创建时间排序)，我可以将其读入 numpy float32 数组。我想将这些数组以 C 程序(我无法修改)可以访问的两种不同格式写入单个文件中的文件系统。

第一种格式很简单:每个数组从左到右从上到下依次排列数组的值。 (数组就是这样一个接一个地出现)。我可以用 np.tofile 简单地做到这一点。

第二种格式更复杂:对于每个像素坐标(x，y)，我想将所有图像的相应像素依次写入文件中。我尝试堆叠数组，然后转置结果。但是，当我使用 np.tofile 将其写入文件系统时，该文件包含与第一种格式相同的数据排列。

如何告诉 numpy 重新排列数据？

Answer 1

对于第二种格式，您可以使用 column_stack 后跟 ravel

In [8]: img1 = np.arange(5, dtype='float32')

In [9]: img2 = np.arange(5, dtype='float32')

In [10]: np.column_stack((img1,img2)).ravel()
Out[10]: array([ 0.,  0.,  1.,  1.,  2.,  2.,  3.,  3.,  4.,  4.], dtype=float32)