python - 如何 pickle scipy.stats 分布(不能 pickle instancemethod 对象)

Question

如何保存 scipy.stats 分布？

例如:

a = [scipy.stats.norm(0,1), scipy.stats.norm(0,2)]
with open("distro.pickle", 'w') as f:
    pickle.dump(a, f)

这样做我得到一个 TypeError: can't pickle instancemethod objects

Answer 1

他们不支持 pickle 。 “解决”您的问题的更简单方法是 pickle 参数，并在取消 pickle 时创建一个新对象:

>>> from collections import namedtuple
>>> Norm = namedtuple('Norm', 'mu variance')
>>> def pickle_norm(n):
...     return pickle.dumps(Norm(*n.args))
... 
>>> def unpickle_norm(s):
...     return scipy.stats.norm(*pickle.loads(s))
... 
>>> s = pickle_norm(scipy.stats.norm(5, 10))
>>> d = unpickle_norm(s)