collections.defaultdict

• 通常情况下在遍历一个数组,找到每个数组里面的每个元素的个数时,我们会选择:

bag = {}
for word in words:
    if word not in bag:
        bag[word] = 1
    else:
        bag[word] += 1

• 像上面那样我们每次都要判断字典中是否有该键,有一个更简单的方法就是使用collections.defaultdict类.defaultdict必须得使用一个类型来初始化:

from collections import defaultdict
dic = defaultdict(int)
print dic 
'''
defaultdict(<type 'int'>, {})
'''

• 初始化的时候使用了一个类型,那么在没有这个键的时候,会自动创建一个默认值,如下发现int类型的默认键值是0

d_int = defaultdict(int) print d_int['a']
'''
0
'''

• list的默认键值是空list[]:

d_list = defaultdict(list) print d_list['a']
'''
[]
'''

• str的默认键值是空字符串”

d_str = defaultdict(str) print d_str['a']
'''
''
'''

• 当然也可以使用lambda自定义初始化的键值:

d_lambda = defaultdict(lambda: 5) print d_lambda ['a']
'''
5
'''

• 也可以初始化嵌套的字典:

d_nest = defaultdict(lambda: defaultdict(int))