python - 将 numpy.random.get_state() 写入文件

Question

numpy.random.get_state() 返回一个 tuple。我想将其写入文件，以便以后需要时可以使用相同的状态。

我把它归结为以下内容，但是 csv 抛出一个错误:

import csv
import numpy as np

def write_state():
    with open("state_file", "w") as file:
        csv.register_dialect("custom", delimiter=" ", skipinitialspace=True)
        writer = csv.writer(file, dialect="custom")
        for val in np.random.get_state():
            writer.writerow(val)

回溯:

  File "MCVE.py", line 21, in <module>
    write_state()
  File "MCVE.py", line 11, in write_state
    writer.writerow(val)
_csv.Error: sequence expected

将 np.random.get_state() 写入文件以便 np.random.set_state() 稍后可以使用存储的信息的适当方法是什么？

Answer 1

Python 内置的cPickle.dump/load 可以高效地写入和读取许多对象，例如元组。

写作

from cPickle import dump
import numpy as np

with open('state.obj', 'wb') as f:
    dump(np.random.get_state(), f)

阅读

from cPickle import load
import numpy as np

with open('state.obj', 'rb') as f:
    np.random.set_state(load(f))