在Python中通过文本文件中的索引创建矩阵

Question

我有以下问题，我有一个由另一个程序生成的矩阵并写入txt文件，该文件看起来像这样:(我只包含部分矩阵，总共36行）

 1     1   7.225655E+03
 1     2   0.000000E+00
 1     3   0.000000E+00
 1     4   0.000000E+00
 1     5   2.384466E+04
 1     6   0.000000E+00
 2     1   0.000000E+00
 2     2   7.225655E+03         .
          .
          .
 5     5   4.175514E+06
 5     6   0.000000E+00
 6     1   0.000000E+00
 6     2  -2.829306E+03
 6     3   0.000000E+00
 6     4   0.000000E+00
 6     5   0.000000E+00
 6     6   3.916341E+06

我已经在numpy数组中加载了这个矩阵，但现在我想写一个简单的函数，它根据前两列的索引将值放在一个6x6矩阵中。

Answer 1

如果a是您从文件中读取的树列矩阵，则应执行以下操作：

>>> m = np.zeros((6,6))
>>> for row, col, val in a:
...    m[row - 1, col - 1] = val