在numpy中将文本文件作为矩阵导入

Question

我对python编程很陌生，所以对于大多数专业人士来说，这看起来很容易。我有一个以下格式的文本文件。我只想把数字导入矩阵。这意味着我不需要空格（每行开头也有空格）和数据标签。

 1 1 1 1 1 1 1 data_1
 1 1 1 1 1 1 2 data_2
 1 1 1 1 1 2 1 data_3
 1 1 1 1 1 2 2 data_4
 1 1 1 1 1 3 1 data_5
 1 1 1 1 1 3 2 data_6

Answer 1

使用numpy.loadtxt，它假定默认情况下数据由空白分隔，并使用参数usecols，指定在构建数组时要使用的字段：

In [1]: import numpy as np
In [2]: matrix = np.loadtxt('matrix.txt', usecols=range(7))
In [3]: print matrix

[[ 1.  1.  1.  1.  1.  1.  1.]
 [ 1.  1.  1.  1.  1.  1.  2.]
 [ 1.  1.  1.  1.  1.  2.  1.]
 [ 1.  1.  1.  1.  1.  2.  2.]
 [ 1.  1.  1.  1.  1.  3.  1.]
 [ 1.  1.  1.  1.  1.  3.  2.]]

如果希望矩阵元素是整数，也可以将dtype=int传递到loadtxt。