有没有办法让defaultdict也成为defaultdict的默认值?(即无限级递归defaultdict?)

我希望能够做到:

x = defaultdict(...stuff...)
x[0][1][0]
{}

所以,我可以做x = defaultdict(defaultdict)但这只是第二层:

x[0]
{}
x[0][0]
KeyError: 0

有些食谱可以做到这一点。但是仅仅使用普通的defaultdict参数就可以做到吗?

注意,这是在询问如何做一个无限级递归defaultdict,所以它与Python不同:defaultdict of defaultdict?,这是如何做一个两级defaultdict。

我可能最终会使用串模式,但当我意识到我不知道如何做时,它让我感兴趣。


当前回答

@nucklehead的响应可以扩展到处理JSON中的数组:

def nested_dict(existing=None, **kwargs):
    if existing is None:
        existing = defaultdict()
    if isinstance(existing, list):
        existing = [nested_dict(val) for val in existing]
    if not isinstance(existing, dict):
        return existing
    existing = {key: nested_dict(val) for key, val in existing.items()}
    return defaultdict(nested_dict, existing, **kwargs)

其他回答

对于任意数量的层:

def rec_dd():
    return defaultdict(rec_dd)

>>> x = rec_dd()
>>> x['a']['b']['c']['d']
defaultdict(<function rec_dd at 0x7f0dcef81500>, {})
>>> print json.dumps(x)
{"a": {"b": {"c": {"d": {}}}}}

当然你也可以用lambda来做这个,但是我发现lambda可读性较差。在任何情况下,它看起来是这样的:

rec_dd = lambda: defaultdict(rec_dd)

@nucklehead的响应可以扩展到处理JSON中的数组:

def nested_dict(existing=None, **kwargs):
    if existing is None:
        existing = defaultdict()
    if isinstance(existing, list):
        existing = [nested_dict(val) for val in existing]
    if not isinstance(existing, dict):
        return existing
    existing = {key: nested_dict(val) for key, val in existing.items()}
    return defaultdict(nested_dict, existing, **kwargs)

这里的其他答案告诉您如何创建一个包含“无限多个”defaultdict的defaultdict,但它们未能解决我认为可能是您最初的需求,即简单地拥有一个双深度defaultdict。

你可能一直在寻找:

defaultdict(lambda: defaultdict(dict))

你可能更喜欢这个结构的原因是:

它比递归解决方案更明确,因此读者可能更容易理解。 这使得defaultdict的“叶子”可以是字典以外的东西,例如:defaultdict(lambda: defaultdict(list))或defaultdict(lambda: defaultdict(set))

这里有一个妙招:

tree = lambda: defaultdict(tree)

然后你可以用x = tree()来创建你的x。

这里有一个类似于@Stanislav的答案的解决方案,适用于多处理,也允许终止嵌套:

from collections import defaultdict
from functools import partial

class NestedDD(defaultdict):
    def __init__(self, n, *args, **kwargs):
        self.n = n
        factory = partial(build_nested_dd, n=n - 1) if n > 1 else int
        super().__init__(factory, *args, **kwargs)

    def __repr__(self):
        return repr(dict(self))

def build_nested_dd(n):
    return NestedDD(n)