5

我正在尝试使用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 = ?我无法通过谷歌弄清楚它来指定多列的格式。

4

2 回答 2

10

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

于 2012-05-25T17:26:30.543 回答
1

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

>>> 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不接受单个整数；它需要一个序列。((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])
于 2012-05-25T17:25:54.417 回答