python - collections模块之defaultdict类

在取dict中值的时候，可以使用setdefault( ) 方法，如果字典中没有key，不会报错，参数二还可以设置默认值，此方法在一定程度上解决了dict中不存在默认值的问题，但是这这时我们想，有没有一种字典它本身提供了默认值的功能？答案是肯定的：collections.defaultdict

 

初始化：

1、defaultdict 类它可以使用类型来初始化字典中的值：

from collections import defaultdict
counts = defaultdict(list)
print(counts['name']) # []

2、defaultdict类的初始化函数接受一个类型作为参数，当访问一个不存在键的时候，可实例化一个值作为默认值：

from collections import defaultdict
counts = defaultdict(list)
print(counts['age']) # []

 

访问：

3、通过defaultdict来初始化，只能通过 dict[key] 或者 dict.__getitem__[key] 来访问，其他方法访问无效：

from collections import defaultdict
counts = defaultdict(list)
print(counts.get('name')) # None
print(counts['name']) # []
print(counts.__getitem__('age')) # []

 

不带参数的函数作为参数：

该类除了接受类型名称作为初始化的函数参数以外，还可以使用不带参数的可调用函数的返回值作为默认值：

from collections import defaultdict
def return_param():
    return [1, 2, 3]
counts = defaultdict(return_param)
print(counts['name']) # [1, 2, 3]

from collections import defaultdict
li = ['a', 'a', 'b']
counts = defaultdict(int)
for key in li:
    counts[key] += 1
# 统计列表中各元素出现的次数
print(dict(counts)) # {'a': 1, 'b': 1}