python – 如何将嵌套字典转换为defaultdict？

2023年2月28日 493次阅读

如何将嵌套字典转换为嵌套的defaultdict？

dic = {"a": {"aa": "xxx"}}
default = defaultdict(lambda: None, dic)
print(default["dummy_key"])  # return None
print(default["a"]["dummy_key"])  # KeyError

最佳答案 您需要循环或递归嵌套字典,通过其所有级别.

除非它可能非常深(如数百个级别),或者如此宽,以至于小的性能因素有所不同,递归在这里可能是最简单的：

def defaultify(d):
    if not isinstance(d, dict):
        return d
    return defaultdict(lambda: None, {k: defaultify(v) for k, v in d.items()})

或者,如果您希望它与所有映射一起使用,而不仅仅是dicts,您可以在isinstance检查中使用collections.abc.Mapping而不是dict.

当然这假设你有一个纯粹的嵌套字典.如果您已经从一个典型的JSON响应中解析了一些内容,那里可能存在带有dict元素的列表值,那么您还必须处理其他可能性：

def defaultify(d):
    if isinstance(d, dict):
        return defaultdict(lambda: None, {k: defaultify(v) for k, v in d.items()})
    elif isinstance(d, list):
        return [defaultify(e) for e in d]
    else:
        return d

但是如果这实际上是来自JSON,那么在解析JSON时,最好只使用你的defaultdict作为object_pairs_hook,而不是将其解析为dict,然后将其转换为defaultdict.

在使用OrderedDict代替dict的文档中有一个例子,但这对我们来说不太适用 – 与OrderedDict和dict不同,defaultdict不能只将一对迭代的对作为其唯一的参数;它首先需要默认值工厂.所以我们可以使用functools.partial绑定它：

d = json.loads(jsonstring, object_hook_pairs=partial(defaultdict, lambda: None))

等等.

点赞