【Python】 探索Python中的`collections.defaultdict`与普通字典的区别

基本原理

在Python中，字典是一种非常灵活的数据结构，用于存储键值对。然而，在某些情况下，我们可能会遇到一些特定的需求，比如希望字典在访问不存在的键时自动提供一个默认值，或者希望字典能够存储多个值。这就是collections模块中的defaultdict大显身手的地方。

defaultdict是Python标准库collections模块提供的一个子类，它继承自Python的内置字典类型。defaultdict的主要特点是它在初始化时需要提供一个工厂函数，当字典中没有对应的键时，就会调用这个工厂函数来生成默认值。

与之相对的，普通字典（dict）在访问不存在的键时会抛出KeyError异常，除非我们显式地使用dict.get()方法或者dict.setdefault()方法，这些方法允许我们为不存在的键提供一个默认值。

代码示例

示例1：基本使用

from collections import defaultdict

# 初始化一个defaultdict，当访问不存在的键时，返回0
dd = defaultdict(int)
dd['a'] = 1
print(dd['a'])  # 输出: 1
print(dd['b'])  # 输出: 0，因为'b'不存在，int()返回0

示例2：存储多个值

# 使用list作为默认工厂函数，可以存储多个值
dd_list = defaultdict(list)
dd_list['a'].append(1)
dd_list['a'].append(2)
print(dd_list['a'])  # 输出: [1, 2]
print(dd_list['b'])  # 输出: [], 因为'b'不存在，list()返回空列表

示例3：自定义默认值

# 使用自定义的工厂函数
def default_factory():
    return "默认值"

dd_custom = defaultdict(default_factory)
print(dd_custom['c'])  # 输出: 默认值

注意事项

1. 使用defaultdict时，确保提供的工厂函数是无参数的，因为它会在字典中不存在键时自动调用。
2. 虽然defaultdict在某些情况下非常有用，但在不需要默认值的情况下，使用普通字典可能更加直观和简单。
3. 记住，defaultdict只是普通字典的一个变体，它也支持字典的所有操作和方法。

结论

collections.defaultdict提供了一种方便的方式来处理字典中不存在的键，通过自动生成默认值，使得代码更加简洁和易于维护。然而，选择使用defaultdict还是普通字典，应根据具体的应用场景和需求来决定。理解defaultdict的工作原理和适用场景，将有助于编写更加高效和灵活的Python代码。

>
> 【痕迹】QQ+微信朋友圈和聊天记录分析工具1.0.4 （1）纯Python语言实现，使用Flask后端，本地分析，不上传个人数据。
>
> （2）内含QQ、微信聊天记录保存到本地的方法，真正实现自己数据自己管理。
>
> （3）数据可视化分析QQ、微信聊天记录，提取某一天的聊天记录与大模型对话。
>
> 下载地址：https://www.alipan.com/s/x6fqXe1jVg1
>