python - 如何将不同维度的numpy数组附加到python中已有的文本文件中

Question

我有一个代码，每次迭代都需要将 2xN 维矩阵写入文本文件。但问题是 N 从一次迭代到另一次迭代不断变化。

我想将此数据写入Python中的文本文件，并继续将新生成的矩阵附加到旧文件中。但似乎我必须先将 numpy 数组转换为字符串，然后才能附加到文本文件，并且它非常耗时，因为大型矩阵的数组到字符串函数非常慢。

因此，有人可以告诉我如何将 numpy 数组直接附加到文本文件中，而不将它们转换为字符串或列表/

谢谢

Answer 1

您可以对不同大小的数组多次使用savetxt:

In [34]: f = open('test.txt', 'wb')  # python3, write bytes not unicode
In [35]: np.savetxt(f,np.ones((2,3),dtype=int))
In [36]: np.savetxt(f,np.ones((2,2),dtype=int))
In [37]: np.savetxt(f,np.ones((2,1),dtype=int))
In [38]: f.close()
In [39]: cat test.txt
1.000000000000000000e+00 1.000000000000000000e+00 1.000000000000000000e+00
1.000000000000000000e+00 1.000000000000000000e+00 1.000000000000000000e+00
1.000000000000000000e+00 1.000000000000000000e+00
1.000000000000000000e+00 1.000000000000000000e+00
1.000000000000000000e+00
1.000000000000000000e+00

如果您希望每次都打开和关闭文件，请使用“a”追加模式。

我不知道 np.savetxt 的速度与您尝试过的相比如何。 “文本”==“字符串”。必须有人将数字数据转换为字符串。

在 savetxt 中，这是逐行完成的:

for row in X:
   fh.write(asbytes(format % tuple(row) + newline))

有一些快速方法可以将 numpy 数组写入二进制数据，但这些不会是“文本”文件。

`