Python collections.defaultdict() 的使用

在Python里面有一个模块collections，解释是数据类型容器模块。这里面有一个collections.defaultdict() 经常被用到。

defaultdict(function_factory)构建的是一个类似dictionary的对象，其中keys的值，自行确定赋值，但是values的类型，是function_factory的类实例，而且具有默认值。比如default(int)则创建一个类似dictionary对象，里面任何的values都是int的实例，而且就算是一个不存在的key, d[key] 也有一个默认值，这个默认值是int()的默认值0。

from collections import defaultdict
s = "aabbcccbad"
d = defaultdict(int)
for i in s:
    d[i] +=1

print(d.items())
dict_items([('c', 3), ('d', 1), ('b', 3), ('a', 3)])

print(d)
defaultdict(<class 'int'>, {'c': 3, 'd': 1, 'b': 3, 'a': 3})

print(list(d.items()))
[('c', 3), ('d', 1), ('b', 3), ('a', 3)]

from operator import itemgetter
m = sorted(d.items(),key = itemgetter(1),reverse = True)
print(m)
[('c', 3), ('b', 3), ('a', 3), ('d', 1)]

defaultdict(int) 这里的d其实是生成了一个默认为0的带key的数据字典。你可以想象成 d[key] = int default （int工厂函数的默认值为0）。

d[k]所以可以直接读取 d[“m”] += 1 就是d[“m”] 就是默认值 0+1 = 1。

defaultdict也可以接受一个内建函数list作为参数。其实，list()本身是内建函数，但是再经过更新后，python里面所有东西都是对象，所以list改编成了类，引入list的时候产生一个类的实例。

s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red',1)]
d = defaultdict(list)
for k, v in s:
    d[k].append(v)
list(d.items())
[('blue', [2, 4]), ('yellow', [1, 3]), ('red', [1])]