Python每日一练0007

问题

怎样实现一个键对应多个值的字典（也叫 multidict）？

例如：

d = {
    'a' : [1, 2, 3],
    'b' : [4, 5]
}
e = {
    'a' : {1, 2, 3},
    'b' : {4, 5}
}

解决方案

使用collections模块中的defaultdict类

defaultdict的特点是，当key不存在时，会自动创建它，也就是说我们不用再像dict一样首先判断key是否存在了，下面是一个例子：

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> d['a'].append(1)
>>> d['a'].append(2)
>>> d['a'].append(3)
>>> d['b'].append(4)
>>> d['b'].append(5)
>>> print(d)
defaultdict(<class 'list'>, {'a': [1, 2, 3], 'b': [4, 5]})

当然，也可以用dict的setdefault函数来实现同样的效果，但看起来会比较“丑陋”

>>> d = dict()
>>> d.setdefault('a', []).append(1)
>>> d.setdefault('a', []).append(2)
>>> print(d)
{'a': [1, 2]}

讨论

根据Python3官方文档对defaultdict的介绍可以了解到

defaultdict是dict的子类，defaultdict重写了dict的__missing__方法，并且增加了一个可以修改的default_factory成员

当defaultdict的key不存在时，会将key初始化为调用default_factory得到的返回值，default_factory可以在初始化时指定，也可以随后再进行改变

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> d['l'].append(1)
>>> d.default_factory = set
>>> d['s'].add(2)
>>> print(d)
defaultdict(<class 'set'>, {'l': [1], 's': {2}})

来源

Python Cookbook

关注

欢迎关注我的微信公众号：python每日一练