python - 读取元组并将其写入文件(numpy.random.RandomState.get_state)

Question

我试图将代码中随机数生成器的状态保存到文件中，然后再将其读回。生成的元组的形式为:

tuple(str, 624 uints 的 ndarray, int, int, float)

返回的元组包含以下项目:

1. 字符串“MT19937”。
2. 由 624 个无符号整数键组成的一维数组。
3. 整数 pos。
4. 整数 has_gauss。
5. float cached_gaussian。

有没有一种简单的方法可以将其写入文件，然后以最初生成的格式读回？

这是我到目前为止所拥有的:

 fx = open('state.dat', 'w')
 fx.write(' '.join(str(s) for s in state) + '\n')
 fx.close()

这确实打印到文件，但我不确定如何将字符串输出转换回原始格式。

非常感谢

瑞恩

Answer 1

您可以使用pickle将磁盘上的任何对象保存为二进制:

import numpy
import pickle

RANDOM_STATE_FILE = 'random.bin'

def save_random_state():
    with open(RANDOM_STATE_FILE, 'wb') as f:
        pickle.dump(numpy.random.RandomState.get_state(), f)

def load_random_state():
    with open(RANDOM_STATE_FILE, 'rb') as f:
        numpy.random.RandomState.set_state(pickle.load(f))