如何将 defaultdicts [of defaultdicts] 的 defaultdict 转换为 dicts [of dicts] 的 dict？

Question

使用this answer，我创建了defaultdict 的defaultdicts。现在，我想把那个嵌套很深的 dict 对象变回一个普通的 python dict。

from collections import defaultdict

factory = lambda: defaultdict(factory)
defdict = factory()
defdict['one']['two']['three']['four'] = 5

# defaultdict(<function <lambda> at 0x10886f0c8>, {
#             'one': defaultdict(<function <lambda> at 0x10886f0c8>, {
#                 'two': defaultdict(<function <lambda> at 0x10886f0c8>, {
#                     'three': defaultdict(<function <lambda> at 0x10886f0c8>, {
#                         'four': 5})})})})

我认为这不是正确的解决方案：

import json

regdict = json.loads(json.dumps(defdict))

# {u'one': {u'two': {u'three': {u'four': 5}}}}

另外，this answer 也不够用，因为它不会在嵌套的 dict(s) 上递归。

Answer 1

您可以在树上递归，将每个 defaultdict 实例替换为由 dict 理解生成的 dict：

def default_to_regular(d):
    if isinstance(d, defaultdict):
        d = {k: default_to_regular(v) for k, v in d.items()}
    return d

演示：

>>> from collections import defaultdict
>>> factory = lambda: defaultdict(factory)
>>> defdict = factory()
>>> defdict['one']['two']['three']['four'] = 5
>>> defdict
defaultdict(<function <lambda> at 0x103098ed8>, {'one': defaultdict(<function <lambda> at 0x103098ed8>, {'two': defaultdict(<function <lambda> at 0x103098ed8>, {'three': defaultdict(<function <lambda> at 0x103098ed8>, {'four': 5})})})})
>>> default_to_regular(defdict)
{'one': {'two': {'three': {'four': 5}}}}

Answer 2

你的actually trying to do 是腌制你的递归defaultdict。而且你不在乎你在 unpickling 时得到的是 dict 还是 defaultdict。

虽然有很多方法可以解决这个问题（例如，使用自己的酸洗创建一个 defaultdict 子类，或者使用 copyreg 显式覆盖默认的子类），但有一种方法非常简单。

请注意您在尝试时遇到的错误：

>>> pickle.dumps(defdict)
PicklingError: Can't pickle <function <lambda> at 0x10d7f4c80>: attribute lookup <lambda> on __main__ failed

您不能腌制lambda 定义的函数，因为它们是匿名的，这意味着它们永远不可能被解封。

但实际上没有理由需要由lambda 定义此函数。特别是，您甚至希望它是匿名的，因为您明确地给它一个名字。所以：

def factory(): return defaultdict(factory)

你已经完成了。

它在行动：

>>> from collections import defaultdict
>>> def factory(): return defaultdict(factory)
>>> defdict = factory()
>>> defdict['one']['two']['three']['four'] = 5
>>> import pickle
>>> pickle.dumps(defdict)
b'\x80\x03ccollections\ndefaultdict\nq\x00c__main__\nfactory\nq\x01\x85q\x02Rq\x03X\x03\x00\x00\x00oneq\x04h\x00h\x01\x85q\x05Rq\x06X\x03\x00\x00\x00twoq\x07h\x00h\x01\x85q\x08Rq\tX\x05\x00\x00\x00threeq\nh\x00h\x01\x85q\x0bRq\x0cX\x04\x00\x00\x00fourq\rK\x05ssss.'

在其他情况下，无缘无故使用lambda 而不是def 会导致问题——你不能在运行时内省你的函数，你会在调试器中得到更糟糕的回溯，等等。使用@987654335 @ 当你想要一个固有的匿名函数，或者你可以在表达式中间定义的函数，但不要用它来节省输入的三个字符。