python numpy savetxt

Question

有人可以指出我在這裏做錯了嗎？

import numpy as np 

a = np.array([1,2,3,4,5],dtype=int) 
b = np.array(['a','b','c','d','e'],dtype='|S1') 

np.savetxt('test.txt',zip(a,b),fmt="%i %s") 

輸出是：

Traceback (most recent call last): 
    File "loadtxt.py", line 6, in <module> 
    np.savetxt('test.txt',zip(a,b),fmt="%i %s") 
    File "/Users/tom/Library/Python/2.6/site-packages/numpy/lib/io.py", line 785, in savetxt 
    fh.write(format % tuple(row) + '\n') 
TypeError: %d format: a number is required, not numpy.string_ 

Answer 1

你需要以不同的方式構建你數組：

z = np.array(zip([1,2,3,4,5], ['a','b','c','d','e']), dtype=[('int', int), ('str', '|S1')]) 
np.savetxt('test.txt', z, fmt='%i %s') 

當你傳遞一個序列，savetext執行asarray(sequence) call並且結果數組的類型爲|S4，即所有元素都是字符串！這就是爲什麼你看到這個錯誤。

Answer 2

我想你所遇到的問題是要傳遞的元組通過格式化字符串，它不能與％I解釋的元組。嘗試使用FMT =「％s」時，假設這是你在找什麼作爲輸出：

1 a 
2 b 
3 c 
4 d 
5 e 

Answer 3

如果你想保存一個CSV文件，您還可以使用該功能rec2csv（包含在matplotlib.mlab）

>>> from matplotlib.mlab import rec2csv 
>>> rec = array([(1.0, 2), (3.0, 4)], dtype=[('x', float), ('y', int)]) 
>>> rec = array(zip([1,2,3,4,5], ['a','b','c','d','e']), dtype=[('x', int), ('y', str)]) 
>>> rec2csv(rec, 'recordfile.txt', delimiter=' ') 

希望，有一天pylab的開發商將實行寫作的CSV文件一個體面的支持。