【参考资料】
1.简书: Python中collections.defaultdict()使用
2.博客园: Python 3 collections.defaultdict() 与 dict的使用和区别
3.官方文档
Backgroud
在使用 python 自带的字典 dict()
时常会碰到比较麻烦的地方, 譬如当 ‘key1’ 不存在时, 使用 D[‘key1’]
会导致 KeyError. 因此引入 collections.defaultdict()
How to use
使用如下代码声明变量
from collections import defaultdict
D = defaultdict(p1)
参数 p1
有比较多的选项,常见的有
1 int, dict, set, str, list
使用这些参数时, 将规定 defaultdict 中 value 的默认数值类型, 也就是说, 当 ‘key1’ 不存在时, D[‘key1’]
将会返回对应的空值 (int: 0, dict: {}, list: ()). 例如
Dd = defaultdict(dict)
# Dd['key1'] = {'key1.1': 'value1.1'} 用这个也可以
Dd['key1']['key1.1'] = 'value1.1'
defaultdict(dict, {‘key1’: {‘key1.1’: ‘value1.1’}})
Dl = defaultdict(list)
Dl['l1'] = [1, 2, 3]
defaultdict(list, {‘l1’: [1, 2, 3]})
Dd['key2']
{}
Dl['l2']
()
需要注意的是, 虽然这些参数不是强制的规定, 譬如完全可以在 defaultdic(str)
中储存 int
, 但是不推荐这样做
2 lambda: None
该参数表示当 ‘key1’ 不存在时, 使用 D[‘key1’]
什么都不会返回 (可以把 None 更换为任意参数, 例如 lambda: 0
表示返回数值 0)
D = defaultdict(lambda: None)
D['key1']
啥都不返回, 此时 D = defaultdict(<function __main__.<lambda>>, {'key1': None})
D = defaultdict(lambda: 0)
D['key1']
0 (参数 lambda: 0
的效果近似于 int
)
Examples:
计算列表中所有单词的出现次数
from collections import defaultdict
words = ['hello', 'world', 'nice', 'world']
# 使用 lambda 来定义简单的函数
# 当然这里直接使用 defaultdict(int) 效果是一样的
counter = defaultdict(lambda: 0)
for kw in words:
counter[kw] += 1
defaultdict(<function main.>, {‘hello’: 1, ‘nice’: 1, ‘world’: 2})