Python中的defaultdict()函数：更轻松地处理字典操作

在Python中，defaultdict()函数是collections模块中的一个内置函数，用于创建一个字典对象，可以提供默认的值类型。

通常情况下，当我们使用字典时，如果访问一个不存在的键，会抛出KeyError异常。但是，使用defaultdict()函数创建的字典，在访问一个不存在的键时，会返回一个默认值，而不是抛出异常。这在处理字典操作时非常有用。

下面是defaultdict()函数的语法：

defaultdict(default_factory)

其中，default_factory是一个可选参数，用于指定默认值的类型。如果不指定，默认为None。

下面是几个使用defaultdict()函数的例子，帮助理解它的用法：

1. 计数器

from collections import defaultdict

# 创建一个defaultdict对象，值的类型为int，默认值为0
counter = defaultdict(int)

# 对列表中的元素进行计数
lst = [1, 2, 3, 2, 1, 3, 4, 5, 2, 1, 3, 4]
for ele in lst:
    counter[ele] += 1

# 输出计数结果
print(counter)

输出结果：

defaultdict(<class 'int'>, {1: 3, 2: 3, 3: 3, 4: 2, 5: 1})

2. 分组

from collections import defaultdict

# 创建一个defaultdict对象，值的类型为列表，默认值为空列表
groups = defaultdict(list)

# 根据首字母进行分组
words = ['apple', 'banana', 'cat', 'dog', 'elephant']
for word in words:
    groups[word[0]].append(word)

# 输出分组结果
print(groups)

输出结果：

defaultdict(<class 'list'>, {'a': ['apple'], 'b': ['banana'], 'c': ['cat'], 'd': ['dog'], 'e': ['elephant']})

3. 字典嵌套

from collections import defaultdict

# 创建一个defaultdict对象，值的类型为字典，默认值为一个空字典
nested_dict = defaultdict(dict)

# 添加键值对到嵌套字典中
nested_dict['a']['x'] = 1
nested_dict['b']['y'] = 2

# 输出嵌套字典
print(nested_dict)

输出结果：

defaultdict(<class 'dict'>, {'a': {'x': 1}, 'b': {'y': 2}})

通过使用defaultdict()函数，我们可以更轻松地处理字典操作，省去了对键是否存在的判断。它在需要处理字典和其他数据结构的问题时非常有用。