numpy.loadtxt 给出“不可迭代”错误

Question

我正在尝试使用numpy.loadtxt 读取文件中的数据，如下所示：

## 14 line of header
3 0 36373.7641026
3 1 36373.7641026
3 2 36373.7641026
...

当我给它这个时：

>>> chunk, power = numpy.loadtxt(bf,skiprows=14,usecols=(1,2),unpack=True)

甚至这个：

>>> power = numpy.loadtxt(bf,skiprows=14,usecols=(2))

上面写着TypeError: 'int' object is not iterable

我认为这是因为前两列显然是整数而不是浮点数，但现在我什至不确定它指的是哪个 int 对象，因为它甚至不会只读取浮点数。如何让loadtxt 工作？

相关：如何使用dtype = ? 指定多列的格式我无法通过 google 弄清楚。

Answer 1

在您的第二个示例中，问题可能出在usecols=(2)。 usecols 必须是一个序列。 (2) 是整数 2，而不是包含 2 的单元素元组，这很可能是错误消息所抱怨的：loadtxt() 正在尝试迭代 int。使用(2,)（或[2]，如果您愿意）。

Answer 2

在这种情况下，很难知道是什么导致了问题，因为您没有向我们提供足够的信息。鉴于您在此处发布的内容，您的代码应该可以工作：

>>> with open('beamtest.out', 'r') as f:
...     f.readlines()
... 
['header 0\n', 'header 1\n', 'header 2\n', 'header 3\n', 'header 4\n', 
 'header 5\n', 'header 6\n', 'header 7\n', 'header 8\n', 'header 9\n', 
 'header 10\n', 'header 11\n', 'header 12\n', 'header 13\n', 
 '3 0 36373.7641026\n', '3 1 36373.7641026\n', '3 2 36373.7641026']
>>> chunk, power = numpy.loadtxt('beamtest.out', skiprows=14,
                                 usecols=(1,2), unpack=True)
>>> chunk
array([ 0.,  1.,  2.])
>>> power
array([ 36373.7641026,  36373.7641026,  36373.7641026])

当然，正如kindall 的回答所示，您的第二个示例将失败，因为usecols 不接受单个整数；它requires a sequence。 （(1) 只是括号中的1；要创建一个元组，您需要在其中添加一个逗号——(1,)。）

下面是一个如何使用dtype指定多列格式的例子：

>>> record = numpy.loadtxt('beamtest.out', skiprows=14, usecols=(1, 2), 
                           dtype={'names':('chunk', 'power'), 
                                  'formats':('i8', 'f8')}) 
>>> record
array([(0, 36373.7641026), (1, 36373.7641026), (2, 36373.7641026)], 
      dtype=[('chunk', '<i8'), ('power', '<f8')])
>>> record['chunk']
array([0, 1, 2])
>>> record['power']
array([ 36373.7641026,  36373.7641026,  36373.7641026])