在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。