在单独的列中将多个 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=',')