Python 给字典生成默认值

Question

使用 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']