如何腌制使用 lambda 函数的 defaultdict？

Question

如何创建函数的pickle文件（defaultDict）？我得到的错误是 不能pickle function objects

from collections import defaultdict
dtree = lambda: defaultdict(tree)

try:    import cPickle as pickle
except: import pickle

#Create defaultdict  object:
hapPkl = dtree()

#Create Pickle file
f  = open("hapP.pkl","wb")
pickle.dump(hapPkl,f)
f.close()

堆栈跟踪：

TypeError                                 Traceback (most recent call last)
<ipython-input-287-376cac3b4f0d> in <module>()
      1 f  = open("hapP.pkl","wb")
----> 2 pickle.dump(hapPkl,f)
      3 f.close()

/usr/lib64/python2.7/copy_reg.pyc in _reduce_ex(self, proto)
     68     else:
     69         if base is self.__class__:
---> 70             raise TypeError, "can't pickle %s objects" % base.__name__
     71         state = base(self)
     72     args = (self.__class__, base, state)

TypeError: can't pickle function objects

Answer 1

cPickle 错误信息有点误导； pickle 版本更好。不是你不能腌制功能；这是他们需要通过__name__ 提供服务。 lambda 将__name__ 设置为'<lambda>'，因此它不可腌制。用def定义它：

def tree():
    return defaultdict(tree)

而且它是可腌制的。 （当你解开它时，你仍然需要一个匹配的 tree 定义。）

Answer 2

一个简单的解决方法是在没有defaultdict 的情况下以不同的方式实现您的树数据结构：

class DTree(dict):
    def __missing__(self, key):
        value = self[key] = type(self)()
        return value

try:    import cPickle as pickle
except: import pickle

#Create dtree object:
hapPkl = DTree()

#Create Pickle file
f = open("hapP.pkl", "wb")
pickle.dump(hapPkl, f)
f.close()