2

我正在使用 python 进行数据分析,并希望找到一种将类的实例保存到文件的好方法,以便以后可以在保持其属性值的情况下重新加载它。特别是,此类的实例将加载作为其他类实例的对象,然后执行需要一段时间才能完成的处理器密集型比较。

例如:

import pickle

class MyClass():
    def __init__(self):
        self.info = OtherClass(option=1)

    def pickle(self):
        f = file('test_file', 'wb')
        pickle.dump(self, f, pickle.HIGHEST_PROTOCOL)
        f.close()

    def unpickle(self):
        f = file('test_file', 'rb')
        pickle.load(f)
        f.close()


class OtherClass():
    def __init__(self, option):
        self.property = option * 2


mydata = MyClass(option=5)

mydata.pickle()

然后稍后导入此模块并运行 mydata.unpickle() 以获取 mydata (我还有一个选项可以查找文件之前是否已被腌制,如果有,则暂停计算和调用 OtherClass 实例)。现在,如果我这样做,我可以加载 mydata,但它不再具有 mydata.info 对象。我的理解是这是pickle的故意部分,但是有没有办法完全保存mydata以便我以后可以使用它?

4

1 回答 1

3

您没有对已加载的数据做任何事情。

class MyClass():
    ...
    @staticmethod
    def unpickle():
        with file('test_file', 'rb') as f:
            return pickle.load(f)

data = MyClass.unpickle()
于 2013-06-20T17:02:36.903 回答