Python: 简述数据结构 collections.defaultdict()

【参考资料】
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})