defaultdict
使用python字典时,键不存在时进行索值会报错,但在索值时,我们有时想将不存在的键对应的值设置成空列表[]
,这时候就可以用defaultdict
例如
>>> from collections import defaultdict
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> for k, v in s:
... d[k].append(v)
...
>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
对于不在defaultdict里的键,在第一次调用时,defaultdict会为这个键提供一个对应的默认值,然后将这对键值加入到defaultdict里,而默认值的设置与生成defaultdict时使用的参数有关。
我们也可以自己定制默认值的生成规则,例如我想把默认值都设置成36
>>> from collections import defaultdict
>>> def my_factory():
... return 15
...
>>> d = defaultdict(my_factory)
>>> d['a']
15
>>> d.keys()
['a']