时间:2021-07-01 10:21:17 帮助过:45人阅读
from collections import defaultdict person_by_age = defaultdict(list) for person in persons: d[person.age].append(person.name)
defaultdict 和 dict 的使用方法一样,只有在初始化的时候必须传入一个 callable 的对象 x,当访问某个还不存在的 key 时,会把 value 自动设置成 x()。比如上例中,当第一次访问某个年龄的人 d[person.age] 就会变成 list(), 也就是 []。
当然也可以使用自己定义的 callable 对象,比如:
d = defaultdict(lambda: 0) d["hello"] += 1 # 1 d["a"] # 0
defaultdict 要比 dict.set_default 效率更高,使用起来也更直观和方便。
标准字典包括一个方法setdefault()来获取一个值,如果值不存在则建立一个默认值。defaultdict初始化容器是会让调用者提前指定默认值。
import collections def default_factory(): return 'default value' d = collections.defaultdict(default_factory, foo = 'bar') print 'd:', d print 'foo =>', d['foo'] print 'var =>', d['bar']
只要所有键都有相同的默认值,就可以使用这个方法。
上面的结果是:
d: defaultdict(, {'foo': 'bar'}) foo => bar var => default value