当我们通过键访问字典,如果键不存在将会返回KeyError错误,如何避免这种错误?需要使用collections.defaultdict()。
它重写了方法_missing_(key),增加了一个可写的实例变量default_factory,实例变量default_factory被missing()方法使用,如果该变量存在,则用以初始化构造器,如果没有,则为None。其它的功能和dict一样。
第一个参数为default_factory属性提供初始值,默认为None;其余参数包括关键字参数(keyword arguments)的用法,和dict构造器用法一样。
使用场景一:当我们需要给一个字典append值,如果有这个键就append,如果没有这个键就创建一个list再append,普通写法会十分繁琐,使用collections.defaultdict()会很简单
from collections import defaultdict
s=[('a',1),('b', 2), ('c', 3), ('d', 4), ('a', 5)]
d=defaultdict(list)
for k, v in s:
d[k].append(v)
a=sorted(d.items())
print(a)
# 输出:
# [('a', [1, 5]), ('b', [2]), ('c', [3]), ('d', [4])]
如果把default_factory改为set就可以去重,改为int可以加和