指定默认值
当字典不存在某个键时指定默认值,需结合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