python - 将文本文件导入为 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。