2012-07-26 25 views
5

cómo escribir varias matrices numpy en un archivo csv en varias columnas?Escribir múltiples matrices NumPy en un archivo CSV en columnas separadas?

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) 

Muchas gracias!

Respuesta

8

creo que esto debería hacer lo que quiera:

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

tengo: File "/usr/lib64/python2.4/site-packages/numpy/lib/io.py", línea 542, en savetxt fh.write (formato% tuple (fila) + '\ n') TypeError: argumento flotante requerido – oops

+0

¿Cuáles son los tamaños y tipos de datos de 'arrA',' arrB', y 'arrC'? – user545424

+0

arrA, B, C son ndarray, pero arrA [i, j], arrB [i, j], arrC [i, j] son ​​float64 .. – oops

Cuestiones relacionadas