首页 > 文章列表 > Python怎么使用defaultdict解决字典默认值

Python怎么使用defaultdict解决字典默认值

Python defaultdict
482 2023-04-25

在Python中,defaultdict是一种特殊类型的字典,它可以自动为字典中不存在的键设置默认值。这种类型的字典是在collections模块中定义的。它为我们提供了一个更方便,更容易使用的字典。

1.导入defaultdict

我们需要先导入包含defaultdict的collections模块。可以通过在Python代码的顶部加入以下代码行来实现这一点:

from collections import defaultdict

2.创建defaultdict

创建defaultdict的方法在语法上与创建Python字典是相同的。我们只需要将defaultdict的类型作为构造函数的参数。例如,如果我们要创建一个存储整数的defaultdict,我们可以使用以下代码:

int_dict = defaultdict(int)

如果我们要在defaultdict中存储字符串,可以使用以下代码:

str_dict = defaultdict(str)

在这两种情况下,我们都通过向defaultdict构造函数传递类型来定义了一个新的defaultdict实例。这个类型对于defaultdict的所有键都将设置为相应的默认值。

3.使用defaultdict

当试图访问defaultdict的键时,如果该键不存在,则会自动设置与传递给构造函数的类型相应的默认值。例如,在下面的代码中,我们使用defaultdict存储一个人的年龄,并将默认值设置为0:

age_dict = defaultdict(int)
age_dict['alice'] = 25
age_dict['bob'] = 30
age_dict['charlie']
print(age_dict)

输出将是 {‘alice’: 25, ‘bob’: 30, ‘charlie’: 0} ,当 ‘charlie’ 还没有在 default 字典中时,自动用默认值 0 进行填充。

在这个例子中,我们给默认字典添加了两个键, ‘alice’ 和 ‘bob’ ,并直接查询了另一个键 ‘charlie’ ,这个键不在字典中。由于 ‘charlie’ 不存在于字典中,defaultdict自动地将该键对应的值设置为位于构造函数中传递的相应类型的默认值(在这种情况下是整数0)。我们可以看到,这确实是我们预期的结果。

4.添加defaultdict默认值

我们还可以通过为defaultdict函数提供一个函数作为默认值来定义一个默认值。这个函数将在字典中缺少某个键的情况下调用,并返回所需的默认值。例如,我们可以使用以下代码定义一个函数,该函数以字典中的键作为其默认值:

def return_key():
    return "unknown"

key_dict = defaultdict(return_key)
key_dict['apple'] = 'fruit'
print(key_dict['pear'])
print(key_dict['banana'])

输出将是:

unknown
unknown

在这个例子中,我们定义了一个函数 return_key() ,它简单地返回 “unknown” 。然后我们创建了一个 default 字典,并使用这个函数作为默认值来定义它。当我们尝试访问 “pear” 和 “banana” 这两个键时,我们得到了 “unknown” 这个默认值。但是,当我们在字典中添加了键 “apple” 时,我们得到了 “fruit”。