这篇文章给大家介绍Python 中的 defaultdict 数据类型,具体内容如下所示:
官方网址
首先,defaultdict
是 dict
的一个子类。通常 Python 中字典(dict)这种数据类型是通过键值对来存取的,当索引一个不存在的键时,就会引发 keyerror
异常。那么,defaultdict
就可以解决这个问题,它可以实现为不存的键值返回一个默认值。
defaultdict
是 collections
包下的一个模块,defaultdict
在初始化时可以提供一个 default_factory 的参数,default_factory 接收一个工厂函数作为参数, 可以是 int、str、list 等内置函数,也可以是自定义函数。
用法实例如下:
示例一: 统计字符串中字母出现的个数。
from collections import defaultdict s = 'mississippi' d = defaultdict(int) for k in s: d[k] += 1 print(d) # 输出 # defaultdict(<class 'int'>, {'m': 1, 'i': 4, 's': 4, 'p': 2})
如果不用 defaultdict 的话,写法如下:
s = 'mississippi' d = {} for k in s: if k not in d: d[k] = 1 else: d[k] = d[k] + 1 print(d) # 输出 # {'m': 1, 'i': 4, 's': 4, 'p': 2}
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)] d = defaultdict(list) for k, v in s: d[k].append(v) print(d) # 输出 # defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2, 4], 'red': [1]})
示例二: 使用 lambda 函数。
q = defaultdict(lambda: [0, 0, 0]) print(q['a']) # 输出 # [0, 0, 0]