如何将 numpy.array 写入文件并保留所有数字？

Question

假设我有一个 numpy 数组，其中包含一些从 scipy.optimize.minimize 获得的高精度浮点数，例如：arr = np.array([9.2387213981273981, 0.3219837123801298])。我想以这种精度将这个数组写入文件，如下所示：

filename = 'parameters.py'
f = open(filename, 'w')
f.write('from numpy import *' + '\n')
f.write('arr = ' + repr(arr))
f.close()

然而，这是行不通的，因为 repr(arr) returns 'array([9.2387214 , 0.32198371])'，这里的精度损失很大。解决此问题的一种可能方法是将 arr 转换为列表并将列表写入文件：

L = list(arr)
filename = 'parameters.py'
f = open(filename, 'w')
f.write('L = ' + repr(L))
f.close()

这里，repr(L) returns '[9.238721398127398, 0.3219837123801298]'，所以没有精度损失。我的问题是，如何将数组写入文件而不将其转换为列表，这样就没有精度损失？

Answer 1

您可以使用 np.printoptions 上下文管理器并将 precision 设置为 15:

with np.printoptions(precision=15), open(filename, "w") as fh:
    fh.write("from numpy import *\n")
    fh.write("arr = " + repr(arr))

之后文件看起来像：

from numpy import *
arr = array([9.238721398127398, 0.32198371238013 ])

How can I write a numpy.array to a file and retain all of the digits?