numpy.loadtxt用于一个或多个输入行

Question

我有一个数据文件,可能是一行或多行.我用它读了numpy.loadtxt.这具有使我的单行数据成为标量的功能.这是有问题的,因为我想在读入后使用循环.请参阅下面的示例

$ cat file1
1
$ cat file2
1
2
$ python --version
Python 2.7.6
$ python 
$ python temp.py 
1.0
2.0
Traceback (most recent call last):
  File "temp.py", line 9, in <module>
    for x in data1:
TypeError: iteration over a 0-d array

码

import numpy

data1=numpy.loadtxt ( 'file1', unpack=True )
data2=numpy.loadtxt ( 'file2', unpack=True )

for x in data2:
    print x

for x in data1:
    print x

我也尝试了相关问题的解决方案:numpy loadtxt单行/行作为列表但我不能让它工作

我补充道

data1 = data1 if usi.shape else [data1]

然而,

$ python temp.py 
Traceback (most recent call last):
  File "temp.py", line 7, in <module>
    data1 = data1 if usi.shape else [data1]
NameError: name 'usi' is not defined

我也尝试过import usi,但是我的系统上没有安装,这对于这么简单的任务来说似乎有点过头了.

我究竟做错了什么？我觉得解决方案很简单,但我找不到http://docs.scipy.org/doc/numpy/reference/generated/numpy.loadtxt.html的提示.

Answer 1

尝试:

data1=numpy.loadtxt ( 'file1', unpack=True, ndmin = 1)