我有一组同时包含字符串和浮点数的列表,例如:
import numpy as num
NAMES = num.array(['NAME_1', 'NAME_2', 'NAME_3'])
FLOATS = num.array([ 0.5 , 0.2 , 0.3 ])
DAT = num.column_stack((NAMES, FLOATS))
我想将这两个列表堆叠在一起,并以列的形式将它们写入文本文件;因此,我想使用numpy.savetxt(如果可能)来执行此操作。
num.savetxt('test.txt', DAT, delimiter=" ")
当我这样做时,会得到以下错误:
>>> num.savetxt('test.txt', DAT, delimiter=" ")
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/Library/Python/2.7/site-packages/numpy-1.8.0.dev_9597b1f_20120920-py2.7-macosx-10.8-x86_64.egg/numpy/lib/npyio.py", line 1047, in savetxt
fh.write(asbytes(format % tuple(row) + newline))
TypeError: float argument required, not numpy.string_
理想的输出文件如下:
NAME_1 0.5
NAME_2 0.2
NAME_3 0.3
如何将字符串和浮点数同时写入文本文件,可能避免使用csv(如果其他人可以阅读,我希望这样做)?有没有其他方法来代替使用numpy.savetxt?
您必须在
savetxt
中指定数据的格式(fmt
),在本例中为字符串(%s
):默认格式是float,这就是它希望使用float而不是字符串的原因,并解释了错误消息。
当前接受的答案实际上并没有解决这个问题,即如何保存同时包含字符串和浮点数的列表。为了完整起见,我提供了一个完整的工作示例,它基于@joris comment中给出的链接,并做了一些修改。
更新:在创建
ab
structured array字节字符串而不是'S6'
字节字符串时,使用'U6'
Unicode字符串dtype,这个示例在Python 3中也可以正常工作。后者在Python2.7中可以工作,但在Python3中可以编写类似b'NAME_1'
的字符串。相关问题 更多 >
编程相关推荐