[英]How to use python numpy.savetxt to write strings and float number to an ASCII file?
我有一组包含字符串和浮点数的列表,例如:
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
),在这种情况下, savetxt
其指定为字符串( %s
):
num.savetxt('test.txt', DAT, delimiter=" ", fmt="%s")
默认格式是浮点数,这就是它期望浮点数而不是字符串的原因,并说明了错误消息。
当前接受的答案实际上并未解决该问题,该问题询问如何保存同时包含字符串和浮点数的列表。 为了完整起见,我提供了一个完整的示例,它基于@joris注释中给出的链接进行了一些修改。
import numpy as np
names = np.array(['NAME_1', 'NAME_2', 'NAME_3'])
floats = np.array([ 0.1234 , 0.5678 , 0.9123 ])
ab = np.zeros(names.size, dtype=[('var1', 'U6'), ('var2', float)])
ab['var1'] = names
ab['var2'] = floats
np.savetxt('test.txt', ab, fmt="%10s %10.3f")
更新:此示例在创建ab
结构化数组而不是'S6'
字节字符串时,通过使用'U6'
字符串dtype ,也可以在Python 3中正常工作。 后一个dtype在Python 2.7中可以使用,但是在Python 3中可以写类似b'NAME_1'
字符串。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.