本文详细介绍了在python中如何高效地合并多个字典中具有相同键的列表值。针对传统字典合并方式无法满足列表值合并需求的场景，文章重点阐述了如何利用`collections.defaultdict`结合列表的`extend`方法，以简洁且pythonic的方式实现列表值的累积合并，并提供了清晰的代码示例和专业指导。
合并字典中列表值的挑战
在Python开发中，我们经常会遇到需要合并多个字典的场景。当这些字典拥有相同的键，并且其对应的值是列表类型时，常规的字典合并操作往往无法达到预期效果。例如，考虑以下两个字典：
dict_1 = {"a": ["1"], "b": ["3"]}
dict_2 = {"a": ["2"], "b": ["3"]}我们期望的合并结果是：
new_dict = {'a': ["1", "2"], 'b': ["3", "3"]}然而，如果使用Python 3.5+提供的字典合并语法{**dict_1, **dict_2}，结果将是：
merged_dic = {**dict_1, **dict_2}
# 结果: {'a': ['2'], 'b': ['3']}这种方法只会用dict_2中相同键的值覆盖dict_1中的值，而不是将列表内容进行合并。因此，我们需要一种更智能、更符合Pythonic风格的方法来处理这种特定类型的合并需求。
让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限
立即学习“Python免费学习笔记（深入）”；
解决方案：利用collections.defaultdict
Python标准库中的collections模块提供了一个名为defaultdict的强大工具，它非常适合解决这类问题。defaultdict是dict的一个子类，它重写了__missing__方法。当访问一个不存在的键时，defaultdict会自动调用其构造函数中提供的工厂函数来生成一个默认值，并将其赋给该键。
在本例中，我们可以将list作为工厂函数传递给defaultdict。这意味着，每当我们尝试访问一个尚不存在的键时，defaultdict会自动为该键创建一个空列表。
实现步骤
- 导入defaultdict: 首先从collections模块导入defaultdict。
- 初始化defaultdict: 创建一个新的defaultdict实例，并将其默认值设置为list。
- 遍历字典: 遍历所有需要合并的字典。
- 遍历键值对: 对于每个字典，遍历其所有的键值对。
- 合并列表值: 使用列表的extend()方法将当前字典的列表值添加到defaultdict中对应键的列表中。extend()方法会将一个可迭代对象的所有元素逐个添加到当前列表中，这正是我们所需的行为。与append()不同，append()会将整个列表作为单个元素添加，导致嵌套列表。
示例代码
from collections import defaultdict
# 待合并的字典
dict_1 = {"a": ["1"], "b": ["3"]}
dict_2 = {"a": ["2"], "b": ["3"]}
dict_3 = {"a": ["4", "5"], "c": ["6"]} # 增加一个字典以展示多字典合并和新键处理
# 创建一个defaultdict，默认值为list
new_dict = defaultdict(list)
# 遍历所有待合并的字典
for d in [dict_1, dict_2, dict_3]:
# 遍历当前字典的键值对
for key, value in d.items():
# 使用extend方法将值列表添加到new_dict中对应键的列表中
# 如果key不存在，defaultdict会自动创建空的list
new_dict[key].extend(value)
# 打印结果
print(new_dict)
# 期望输出: defaultdict(<class 'list'>, {'a': ['1', '2', '4', '5'], 'b': ['3', '3'], 'c': ['6']})
# 如果需要标准的dict类型，可以将其转换为dict
final_dict = dict(new_dict)
print(final_dict)
# 期望输出: {'a': ['1', '2', '4', '5'], 'b': ['3', '3'], 'c': ['6']}代码解析
- new_dict = defaultdict(list): 这一行创建了一个特殊的字典。当你尝试访问new_dict['some_key']而'some_key'不存在时，它不会抛出KeyError，而是会自动执行list()并将其结果（一个空列表[]）作为'some_key'的值。
- for d in [dict_1, dict_2, dict_3]:: 这是一个循环，迭代所有需要合并的字典。
- for key, value in d.items():: 在每个字典内部，我们遍历其所有的键和对应的值。
- new_dict[key].extend(value): 这是核心操作。
- 如果key在new_dict中尚不存在，defaultdict会为其创建一个新的空列表。
- 然后，extend(value)方法会将value（它本身是一个列表）中的所有元素添加到new_dict[key]所指向的列表中。例如，如果new_dict[key]是['1']，value是['2']，执行后new_dict[key]会变成['1', '2']。
优点与注意事项
- Pythonic与简洁: 这种方法利用了defaultdict的特性，使得代码逻辑清晰、简洁，符合Python的优雅风格。
- 高效性: 相比于手动检查键是否存在并初始化列表，defaultdict在内部进行了优化，通常效率更高。
- 处理新键: 对于在某些字典中出现而在其他字典中不存在的键，defaultdict能够自动处理，为其创建新的列表并添加值，无需额外的条件判断。
- 可扩展性: 这种方法可以轻松扩展到合并任意数量的字典。只需将所有字典放入一个列表中进行迭代即可。
- extend() vs append(): 务必使用extend()方法。如果使用append()，结果将是列表的列表，例如{'a': [['1'], ['2']]}，这不是我们期望的合并列表元素的效果。
总结
当需要在Python中合并多个字典，并且这些字典的键相同且值是列表类型时，使用collections.defaultdict结合列表的extend()方法是最佳实践。这种方法不仅代码简洁、易于理解，而且能够高效地处理各种合并场景，包括新键的自动创建和多个字典的迭代合并。掌握这一技巧将有助于您编写更健壮、更具Pythonic风格的数据处理代码。
