numpy loadtxt单行/行作为列表

Question

我有一个只有一行的数据文件,如:

 1.2  2.1  3.2

我使用numpy版本1.3.0 loadtxt加载它

 a,b,c = loadtxt("data.dat", usecols(0,1,2), unpack=True)

输出是浮点而不是数组

 a = 1.2

我希望它会是:

 a = array([1.2])

如果我读了一个包含多行的文件,它就可以了.

Answer 1

只需使用 numpy 的 inbuit loadtxt 参数 ndmin 即可。

a,b,c=np.loadtxt('data.dat',ndmin=2,unpack=True)

输出

a=[1.2]

Answer 2

不使用 reshape 的简单方法是显式对列表进行类型转换

 a,b,c = loadtxt("data.dat", usecols(0,1,2), unpack=True)
 a,b,c = (a,b,c) if usi.shape else ([a], [b], [c])

这比重塑更快！