在Python中,字典是一种非常有用的数据结构,它允许我们存储键值对,并且可以快速地通过键来访问相应的值。然而,当我们尝试访问一个不存在的键时,Python会抛出一个KeyError异常。为了解决这个问题,Python提供了Defaultdict类,它允许我们为不存在的键提供一个默认值。

Defaultdict是Python的内建字典子类,它修改了字典的__missing__方法,使得当我们尝试获取不存在的键时,返回一个默认值而不是抛出KeyError异常。这个默认值可以是任何我们想要的类型,包括数字、字符串、列表、字典等。
下面是一个使用Defaultdict的简单示例:
- from collections import defaultdict
-
- # 创建一个Defaultdict,默认值为0
- d = defaultdict(int)
-
- # 添加键值对
- d['a'] = 1
- d['b'] = 2
-
- # 访问不存在的键,输出为0
- print(d['c']) # 输出:0
-
- # 使用另一个默认值创建Defaultdict
- d = defaultdict(str, {'a': 1, 'b': 2})
-
- # 访问不存在的键,输出为空字符串
- print(d['c']) # 输出:''
在上面的示例中,我们首先创建了一个默认值为0的Defaultdict。然后,我们添加了两个键值对。当我们尝试访问一个不存在的键'c'时,输出为0,这是因为我们在创建Defaultdict时指定了默认值为0。接着,我们创建了一个新的Defaultdict,这次我们指定了默认值为空字符串。当我们尝试访问一个不存在的键'c'时,输出为空字符串。
除了使用内置函数来创建Defaultdict之外,我们还可以直接从collections模块导入Defaultdict类并使用它来创建字典。例如:
- from collections import defaultdict
-
- # 创建一个Defaultdict,默认值为列表[]
- d = defaultdict(list)
-
- # 添加键值对
- d['a'].append(1)
- d['b'].append(2)
-
- # 访问不存在的键,输出为空列表[]
- print(d['c']) # 输出:[]
在上面的示例中,我们创建了一个默认值为列表的Defaultdict。然后,我们添加了两个键值对,并尝试访问一个不存在的键'c'。输出结果为空列表,这是因为我们在创建Defaultdict时指定了默认值为列表。
Defaultdict还可以用于更复杂的情况,例如默认值是另一个字典或函数。下面是一个示例,其中默认值是一个空的字典:
- from collections import defaultdict
-
- # 创建一个Defaultdict,默认值为空字典 {}
- d = defaultdict(dict)
-
- # 添加键值对
- d['a']['x'] = 1
- d['b']['y'] = 2
-
- # 访问不存在的键,输出为空字典 {}
- print(d['c']) # 输出:{}
在上面的示例中,我们创建了一个默认值为空字典的Defaultdict。然后,我们添加了两个键值对,并尝试访问一个不存在的键'c'。输出结果为空字典,这是因为我们在创建Defaultdict时指定了默认值为空字典。
此外,Defaultdict还可以使用函数作为默认值。当使用函数作为默认值时,每次访问不存在的键时,都会调用该函数并返回其结果。下面是一个示例,其中默认值是一个计数器函数:
- from collections import defaultdict
-
- # 创建一个Defaultdict,默认值为计数器函数 lambda: 0
- d = defaultdict(lambda: 0)
-
- # 添加键值对
- d['a'] += 1
- d['b'] += 2
-
- # 访问不存在的键,输出为0
- print(d['c']) # 输出:0
在上面的示例中,我们创建了一个默认值为计数器函数的Defaultdict。然后,我们添加了两个键值对,并尝试访问一个不存在的键'c'。输出结果为0,这是因为我们在创建Defaultdict时指定了默认值为计数器函数,它会在每次访问不存在的键时返回0。
使用Defaultdict需要注意以下几点:
default_factory方法来更新默认值。__missing__方法中指定默认值,可以实现类似Defaultdict的行为。总之,Defaultdict是Python中非常有用的数据结构,它允许我们在访问不存在的键时提供一个默认值,而不是抛出KeyError异常。通过使用不同的默认值类型,我们可以将Defaultdict应用于各种不同的场景中。