如何解决如何将 numpy.array 写入文件并保留所有数字?
假设我有一个 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)
返回 '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)
返回 '[9.238721398127398,0.3219837123801298]'
,因此没有精度损失。我的问题是,如何将数组写入文件而不将其转换为列表,以便没有精度损失?
解决方法
您可以使用 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 ])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。