一、defaultdict概述
Python内置的数据类型中,字典是常用的一种。字典的key必须是可哈希的,value可以是任何数据类型。通常在使用字典时,需要先判断一个key是否存在,如果不存在则需要初始化这个key,在设置value。这个过程比较繁琐,如果希望key在不存在的时候直接返回默认值,defaultdict就可以派上用场了。
defaultdict是Python中collections库中的一个类。它初始化时需要一个参数default_factory,可以是任意可调用对象,比如函数、类、lambda表达式等。当字典里访问不存在的key时,default_factory会被调用,其返回值会被赋给这个key。
from collections import defaultdict
d1 = defaultdict(list)
print(d1['a']) # []
d2 = defaultdict(int)
print(d2['a']) # 0
d3 = defaultdict(str)
print(d3['a']) # ''
二、default_factory为list/dict/其他可变对象类型
default_factory常常被用在字典的值是可变对象类型的时候。比如列表、字典等。使用defaultdict可以避免判断key是否存在,然后初始化值。只需要直接操作即可。下面是示例代码。
default_dict = defaultdict(list)
input_list = [('a', 1), ('b', 2), ('a', 3), ('b', 4), ('c', 5)]
# 使用defaultdict
for k, v in input_list:
default_dict[k].append(v)
print(default_dict) # defaultdict(
, {'a': [1, 3], 'b': [2, 4], 'c': [5]})
# 不使用defaultdict
pure_dict = {}
for k, v in input_list:
if k in pure_dict:
pure_dict[k].append(v)
else:
pure_dict[k] = [v]
print(pure_dict) # {'a': [1, 3], 'b': [2, 4], 'c': [5]}
在上面这个例子中,如果直接使用字典,那么每次需要判断key是否存在,并且初始化value类型。如果使用defaultdict,则可以避免这个繁琐的过程。同时,还可以简化代码,提高代码可读性。
三、default_factory为函数
除了使用内置类型,也可以使用函数作为default_factory。这个函数需要返回一个默认的值。
from collections import defaultdict
def get1():
return 1
default_dict_func = defaultdict(get1)
print(default_dict_func['a']) # 1
在上面的代码中,通过定义一个函数get1,将这个函数作为default_factory参数。当defaultdict中不存在key时,就会自动调用get1函数来得到一个默认值,这里是1。
四、default_factory为lambda表达式
lambda表达式也可以作为default_factory参数使用,这个lambda表达式往往比函数更加简洁,也更加直接。
from collections import defaultdict
default_dict_func_lambda = defaultdict(lambda: 0)
print(default_dict_func_lambda['a']) # 0
在上面的代码中,使用lambda表达式构造一个返回0的function。当defaultdict中不存在key时,就会自动调用这个lambda函数来得到默认值,这里是0。
五、defaultdict与Counter结合使用
Counter是Python中collections库中的一个类,它可以自动对一个序列里的元素进行计数。将Counter和defaultdict结合,则可以自动统计一个序列中元素出现的次数。
from collections import defaultdict, Counter
input_list = ['a', 'b', 'c', 'a', 'b', 'a']
cnt = defaultdict(int)
for item in input_list:
cnt[item] += 1
cnt_counter = Counter(input_list)
print(cnt) # defaultdict(
, {'a': 3, 'b': 2, 'c': 1})
print(cnt_counter) # Counter({'a': 3, 'b': 2, 'c': 1})
在上面的代码中,分别使用defaultdict和Counter统计了一个序列中元素出现次数。它们的结果是相同的。但是Counter的代码要更加简洁,而且可以通过most_common函数获取出现次数最多的几项。
六、结语
本文首先介绍了defaultdict的基本用法,然后从default_factory为list、default_factory为函数、default_factory为lambda表达式等几个方面深入讲解了defaultdict的使用方法。最后,介绍了defaultdict和Counter结合使用的方法,希望可以给大家在实际工作中带来帮助。