collections中defaultdict的使用

原文链接：https://blog.csdn.net/yangsong95/article/details/82319675
python通过key访问字典，当key不存在时，会引发‘KeyError’异常。为了避免这种情况的发生，可以使用collections类中的defaultdict()方法来为字典提供默认值。
语法格式： collections.defaultdict([default_factory[, …]])
该函数返回一个类似字典的对象。defaultdict是Python内建字典类（dict）的一个子类，它重写了方法missing(key)，增加了一个可写的实例变量default_factory,实例变量default_factory被missing()方法使用，如果该变量存在，则用以初始化构造器，如果没有，则为None。其它的功能和dict一样。
第一个参数为default_factory属性提供初始值，默认为None,；其余参数包括关键字参数（keyword arguments）的用法，和dict构造器用法一样。

1、使用list作第一个参数，可以很容易将键-值对序列转换为列表字典。

from collections import defaultdict
s=[('yellow',1),('blue',2),('yellow',3),('blue',4),('green',5)]
d= defaultdict(list)
for k,v in s:
    d[k].append(v)
a=sorted(d.items())
print(d,'\n',a)
#输出
defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2, 4], 'green': [5]}) 
 [('blue', [2, 4]), ('green', [5]), ('yellow', [1, 3])]

2、defaultdict还可以被用来计数，将default_factory设为int即可。
字符串中的字母第一次出现时，字典中没有该字母，default_factory函数调用int()为其提供一个默认值0,加法操作将计算出每个字母出现的次数。

s = 'mississippi'
d=defaultdict(int)
for i in s:
    d[i]+=1
a=sorted(d.items())
print(d,'\n',a)
defaultdict(<class 'int'>, {'m': 1, 'i': 4, 's': 4, 'p': 2}) 
 [('i', 4), ('m', 1), ('p', 2), ('s', 4)]

函数int()是常值函数的一种特例，总是返回0。使用匿名函数（lambda function）可以更快、更灵活的创建 常值函数，返回包括0在内的任意常数值。

def constant_factory(value):
    return lambda: value
d= defaultdict(constant_factory('HongKong'))
print(d)
d.update(name='Rick',action='run')
print(d)
print('%(name)s %(action)s to %(object)s'%d)
#输出
defaultdict(<function constant_factory.<locals>.<lambda> at 0x000001D5F3E60D90>, {})
defaultdict(<function constant_factory.<locals>.<lambda> at 0x000001D5F3E60D90>, {'name': 'Rick', 'action': 'run'})
Rick run to HongKong

3、default_factory设为set时，可以用defaultdict建立集合字典（a dictionary of sets）。

s=[('yellow',1),('blue',2),('yellow',3),('blue',4),('green',5)]
d= defaultdict(set)
for k,v in s:
    d[k].add(v)
a=sorted(d.items())
print(d,'\n',a)
#输出
defaultdict(<class 'set'>, {'yellow': {1, 3}, 'blue': {2, 4}, 'green': {5}}) 
 [('blue', {2, 4}), ('green', {5}), ('yellow', {1, 3})]