numpy 我如何将浮点概率存储到一个文件中,使它们的和精确到1?

muk1a3rh  于 2023-01-05  发布在  其他
关注(0)|答案(1)|浏览(198)

我想把一个numpy数组存储到一个文件中。这个数组包含了数千个浮点概率,它们的和都是1。但是当我把这个数组存储到一个CSV文件中并重新加载它时,我意识到这些数字已经被近似了,它们的和现在是0.9999。我该如何修复它呢?
(Numpy的随机选择方法要求概率之和为1)

ilmyapht

ilmyapht1#

请尝试使用np.savetxt

import numpy as np

arr = np.random.random(1000)
arr /= arr.sum()
np.savetxt('arr.csv', arr, delimiter=',')

arr = np.loadtxt('arr.csv')
print(arr.sum())
# >>> 1.0

相关问题