默认情况下的嵌套词典
问题描述
有没有办法创建一个嵌套词典,使我可以说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
是为递增整数的特定目的而创建的,则这将是我更喜欢的方法。
相关文章