collections.defaultdict
的使用方法与普通字典类似,但在访问不存在的键时会自动创建该键,并关联一个指定的默认值。以下是 defaultdict
的基本用法:
-
创建
defaultdict
对象:使用
collections.defaultdict
创建一个默认字典,并指定默认值的类型。在下面的例子中,默认值类型为int
,默认值为0
:from collections import defaultdict # 创建 defaultdict,指定默认值类型为 int,默认值为 0 my_dict = defaultdict(int)
-
访问字典的键:
与普通字典相同,可以通过键访问和操作字典中的值:
my_dict['key1'] += 1 my_dict['key2'] += 2
如果键不存在,
defaultdict
会自动创建该键,并使用默认值:print(my_dict['key3']) # 输出: 0
-
使用自定义默认值:
除了使用内建类型(如
int
)作为默认值类型外,你也可以使用自定义函数作为默认值类型。在下面的例子中,默认值类型是一个返回字符串的 lambda 函数:from collections import defaultdict # 创建 defaultdict,指定默认值类型为 lambda 函数 my_dict = defaultdict(lambda: "default_value") print(my_dict['key1']) # 输出: default_value
这样,如果访问不存在的键,将自动使用 lambda 函数的返回值作为默认值。
defaultdict
主要优势在于避免了在访问字典中不存在的键时引发 KeyError
的问题,而直接提供了默认值,使得代码更简洁。