defaultdict用法

指定默认值

当字典不存在某个键时指定默认值,需结合lambda匿名函数:

from collections import defaultdict
dic = defaultdict(lambda: 0)
print(dic[1]) # 输出 0

注意需指定默认类型或者使用匿名函数。否则报错:

from collections import defaultdict
dic = defaultdict()
print(dic[1]) # 报错 KeyError: 1

嵌套字典

做图相关的题目可能用到:

from collections import defaultdict
dic = defaultdict(lambda:defaultdict(lambda: 0))
print(dic[1][2]) # 输出 0
上一篇:python常用模块-collections模块


下一篇:python 挂钩函数