python - 如何在文本文件中写入 numpy 矩阵 - python

Question

假设我从一些计算中得到一个 numpy 矩阵。这是我的 numpy 矩阵 'result1'::

    result1=
    [[   1.         0.         0.         0.00375   -0.01072   -0.      -1000.     ]
     [   2.         3.         4.         0.        -0.004    750.         0.     ]
     [   3.         3.         0.         0.         0.      -750.      1000.     ]]

现在我想把这个矩阵写在一个名为“result.txt”的文本文件中。为此，我编写了以下代码::

np.savetxt('result.txt', result1, fmt='%.2e')

但它在一行中给出了矩阵的所有元素。

    1.00e+00 0.00e+00 0.00e+00 3.75e-03 -1.07e-02 -1.14e-13 -1.00e+032.00e+00 3.00e+00 4.00e+00 0.00e+00 -4.00e-03 7.50e+02 0.00e+003.00e+00 3.00e+00 0.00e+00 0.00e+00 0.00e+00 -7.50e+02 1.00e+03

我想以适当的矩阵格式在文本文件中写入矩阵。我怎样才能做到这一点？我使用关键字 newline='\n' 或 newline=''，但结果是一样的。

提前致谢...

=======

这个编辑的部分是给@Warren的

试试这个:

>>> import numpy as np
>>> mat=np.matrix([[1, 2, 3],[4, 5, 6],[7, 8, 9]])
>>> mat
matrix([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])
>>> np.savetxt('text.txt',mat,fmt='%.2f')

在我的 text.txt 文件中，我得到:

1.00 2.00 3.004.00 5.00 6.007.00 8.00 9.00

Answer 1

与 Francesco Nazzaro 的回答类似，但为了确保文件可以成功打开，请尝试:

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
mat = np.matrix(a)
with open('outfile.txt','wb') as f:
    for line in mat:
        np.savetxt(f, line, fmt='%.2f')