您的位置:

Python工程师必知:深入解析defaultdict的用法

一、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结合使用的方法,希望可以给大家在实际工作中带来帮助。