2012-07-26 22 views
5

Come scrivere più array NumPy in un unico file CSV in più colonne?Scrivi più array NumPy in file CSV in colonne separate?

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) 

Grazie mille!

risposta

8

penso che questo dovrebbe fare quello che vuoi:

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

ho ottenuto: File "/usr/lib64/python2.4/site-packages/numpy/lib/io.py", la linea 542, in savetxt fh.write (formato% tupla (row) + '\ n') TypeError: argomento float obbligatorio – oops

+0

Quali sono le dimensioni e i tipi di dati di 'arrA',' arrB' e 'arrC'? – user545424

+0

arrA, B, C sono narray, ma arrA [i, j], arrB [i, j], arrC [i, j] sono float64 .. – oops