在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它.