Python pickle 实例变量

Question

我正在对实例变量进行一些计算，完成后我想 pickle 类实例，这样我就不必再次进行计算。这里有一个例子:

import cPickle as pickle

class Test(object):
    def __init__(self, a, b):
        self.a = a
        self.b = b
        self.c = None
    def compute(self, x):
        print 'calculating c...'
        self.c = x * 2

test = Test(10, 'hello')
test.compute(6)

# I have computed c and I want to store it, so I don't have to recompute it again:

pickle.dump(test, open('test_file.pkl', 'wb'))

在 test.compute(6) 之后，我可以查看 test.__dict__ 是什么:

>>> test.__dict__
{'a': 10, 'c': 12, 'b': 'hello'}

我以为这就是要 pickle 的东西；然而，

当我去加载类实例时:

import cPickle as pickle

from pickle_class_object import Test

t2 = pickle.load(open('test_file.pkl', 'rb'))

我在 shell 中看到了这个:

calculating c...

这意味着我没有 pickle c 并且我正在重新计算它。

有没有办法按照我的意愿来 pickle test？所以我不必重新计算 c。我看到我可以 pickle test.__dict__，但我想知道是否有更好的解决方案。此外，我对这里发生的事情的理解很薄弱，所以任何关于正在发生的事情的评论都会很棒。我读过有关 __getstate__ 和 __setstate__ 的内容，但我不知道如何在此处应用它们。

Answer 1

您正在再次导入 pickle_class_object 模块，Python 运行该模块中的所有代码。

您的顶级模块代码包括对 .compute() 的调用，这就是被调用的内容。

您可能希望将创建 pickle 的代码移出 模块，或将其移至 if __name__ == '__main__': protected 部分:

if __name__ == '__main__':
    test = Test(10, 'hello')
    test.compute(6)

    pickle.dump(test, open('test_file.pkl', 'wb'))

仅当作为主脚本运行python文件时，__name__设置为__main__；当作为模块导入时，__name__ 设置为模块名称，if 分支将不会运行。