Python中defaultdict的设置默认值方法

defaultdict是Python内置模块collections中的一个类，可以用来创建一个字典，当访问不存在的键时，返回一个默认值。与普通的dict不同，defaultdict不需要为每个新的键指定默认值，而是在初始化时指定。

一、基本使用

使用defaultdict要先导入collections模块：

from collections import defaultdict

定义一个defaultdict对象：

fruits = defaultdict(int)

这里我们定义了一个名为fruits的defaultdict对象，使用int作为默认值，在访问不存在的键时将返回0。我们可以像使用字典一样使用该对象：

fruits['apple'] += 2
fruits['orange'] += 3

现在fruits字典中已经有了键'apple'和'orange'，并且它们对应的值分别为2和3。

二、自定义默认值函数

除了使用简单的类型值作为默认值，我们还可以使用函数来定义默认值。这个函数会在访问不存在的键时被调用，并且应该返回一个默认值。

比如，我们可以使用自定义函数作为默认值，在访问不存在的键时返回一个空列表：

def create_list():
    return []
fruit_list = defaultdict(create_list)

fruit_list['apple'].append('red')
fruit_list['banana'].append('yellow')

现在fruit_list包含两个键'apple'和'banana'，他们的值分别为['red']和['yellow']。

三、使用lambda表达式设置默认值

除了定义函数，我们还可以使用lambda表达式作为默认值：

fruit_list = defaultdict(lambda: [])

fruit_list['apple'].append('red')
fruit_list['banana'].append('yellow')

这样也能得到和上面的代码一样的效果。

四、嵌套的defaultdict

我们可以在defaultdict中定义一个嵌套的defaultdict。这个功能可以很方便地创建嵌套字典，并在访问不存在的键时自动创建嵌套字典的键。

例如我们可以创建一个名为city_population的字典，其中每个城市都有一个嵌套字典，嵌套字典的键为'year'和'population'：

city_population = defaultdict(lambda: defaultdict(int))

city_population['beijing']['year'] = 2020
city_population['beijing']['population'] = 21.54

city_population['shanghai']['year'] = 2020
city_population['shanghai']['population'] = 24.28

现在city_population中包含两个键'beijing'和'shanghai'，并且它们的值分别是一个嵌套字典，该嵌套字典中包含键'year'和'population'以及对应的值。

五、结语

以上就是defaultdict的几种常用设置默认值的方法，使用defaultdict能够让我们的代码更加简洁明了。除此之外，defaultdict还支持其他一些高级特性，可以在官方文档中查看。