python - 如何腌制 scipy.stats 分布（不能腌制 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

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

>>> 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)