77

我有一组包含字符串和浮点数的列表，例如：

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？

4

2 回答 2

132

您必须在 中指定fmt数据的格式 ( ) savetxt，在本例中为字符串 ( %s)：

num.savetxt('test.txt', DAT, delimiter=" ", fmt="%s")

默认格式是浮点数，这就是它期望浮点数而不是字符串的原因，并解释了错误消息。

于 2013-05-18T08:37:15.770 回答
39

当前接受的答案实际上并未解决该问题，该问题询问如何保存包含字符串和浮点数的列表。为了完整起见，我提供了一个完整的工作示例，该示例基于@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")

更新：此示例在 Python 3 中也可以正常工作，在创建结构化数组'U6'时使用Unicode 字符串dtype ，而不是字节字符串。后一种 dtype 可以在 Python 2.7 中工作，但会像在 Python 3 中那样编写字符串。ab 'S6'b'NAME_1'

于 2016-02-04T18:42:14.873 回答