Geek asks and answers

Write multiple NumPy arrays to a CSV file in separate columns?

How to write multiple numpy arrays in 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)

Thank you so much!

+5
source share
1 answer

I think this should do what you want:

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

More articles:


All Articles