Python pickle实例变量

Question

我正在对一个实例变量进行一些计算，然后完成此操作，我想使该类实例腌制，这样我就不必再次进行计算。 这里是一个例子：

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'}

我以为那会被腌制。 然而，

当我去加载类实例时：

import cPickle as pickle

from pickle_class_object import Test

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

我在外壳中看到了这一点：

calculating c...

这意味着我没有腌制c ，而是在重新计算。

有没有一种方法来腌制test我怎么想？ 因此，我不必再次计算c 。 我看到我可以腌制test.__dict__ ，但是我想知道是否有更好的解决方案。 另外，我对这里发生的事情的了解很薄弱，因此任何有关正在发生的事情的评论都会很棒。 我已经读过有关__getstate__和__setstate__ ，但是我在这里看不到如何应用它们。

Answer 1

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

您的顶级模块代码包括对.compute()调用。

您可能需要将创建泡菜的代码移出模块，或将其移至if __name__ == '__main__':保护的部分：

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

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

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

Answer 2

酸洗的工作原理与您预期的一样。 这里的问题是，当您运行新脚本时，您将导入包含类Test的模块。 整个模块都在运行，包括您创建test 。

处理此类问题的典型方法是在if __name__ == "__main__:块中保护这些内容。

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

if __name__ == "__main__":
    import cPickle as pickle

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

Answer 3

那不是正在发生的事情。 您导入的是其中包含代码的python模块，该模块在顶级时执行。 您可以看到您的代码按预期工作：

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)

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

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


--output:--
calculating c...
12

如果您的代码按您描述的那样工作，那么您将看到两次“计算c ...”。