如何将多个NumPy数组分别写入CSV文件的不同列中？

Question

如何将多个numpy数组写入一个csv文件的多列中？

import numpy
import csv

arrA = numpy.array(file.root.a)
arrB = numpy.array(file.root.b)
arrC = numpy.array(file.root.c)

for i in range (480):
    for j in range (640):
        (write arrA[i,j] into column1,write arrB[i,j] into column2,write arrC[i,j] into column3)

非常感谢！

- oops

Answer 1

我认为这应该可以满足你的需求：

output = np.column_stack((arrA.flatten(),arrB.flatten(),arrC.flatten()))
np.savetxt('output.dat',output,delimiter=',')

- user545424

---

我得到了以下错误信息：File "/usr/lib64/python2.4/site-packages/numpy/lib/io.py", line 542, in savetxt fh.write(format % tuple(row) + '\n') TypeError: 需要浮点数参数。 - oops

arrA、arrB和arrC的大小和数据类型是什么？ - user545424

数组的形状是什么；即arrA.shape是什么？ - user545424

@oops：我更新了我的回答。 np.dstack 给 updated 添加了一个额外的维度。 - user545424