在单独的列中将多个NumPy阵列写入CSV文件？

Question

How to write multiple numpy arrays into one csv file in multiple columns？

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)

Thanks a lot!

Answer 1

I think this should do what you want:

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