python - 将多个 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)

非常感谢!

Answer 1

我认为这应该做你想做的:

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