默认情况下的嵌套词典

2022-04-03 00:00:00 python dictionary counter defaultdict

问题描述

有没有办法创建一个嵌套词典,使我可以说mydict[x][y][z] += 1,其中mydict[x][y][z]以前不存在,默认为0(递增后将为1)?

我研究了一个类似问题的答案,在该问题中,您可以使用collections类(Declaring a multi dimensional dictionary in python)中的defaultdict回答mydict[x][y][z] = 1,但这不允许您采用默认值然后递增。


解决方案

可以,您可以使用collections模块:

from collections import defaultdict, Counter

d = defaultdict(lambda: defaultdict(lambda: Counter()))

d['A']['B']['C'] += 1

# defaultdict(<function __main__.<lambda>>,
#             {'A': defaultdict(<function __main__.<lambda>.<locals>.<lambda>>,
#                          {'B': Counter({'C': 1})})})

注意:也可以只使用嵌套defaultdict

d = defaultdict(lambda: defaultdict(lambda: defaultdict(int)))

但是,如果Counter是为递增整数的特定目的而创建的,则这将是我更喜欢的方法。

相关文章