2012-07-26 72 views
5

如何寫多個numpy的陣列到一個CSV文件中的多個列?在不同的列中將多個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) 

非常感謝!

回答

8

我想這應該做你想要什麼:

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

我:文件 「/usr/lib64/python2.4/site-packages/numpy/lib/io.py」,線路542,在savetxt fh.write(format%tuple(row)+'\ n') TypeError:需要浮點參數 – oops 2012-07-26 03:06:00

+0

'arrA','arrB'和'arrC'的大小和數據類型是什麼? – user545424 2012-07-26 03:34:36

+0

arrA,B,C是ndarray,但是arrA [i,j],arrB [i,j],arrC [i,j]是float64 .. – oops 2012-07-26 03:43:01