Python中合并具有相同键的字典值列表：高效策略与defaultdict应用

本文详细介绍了在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风格的方法来处理这种特定类型的合并需求。

立即学习“Python免费学习笔记（深入）”；

Procys

AI驱动的发票数据处理

102

查看详情

解决方案：利用collections.defaultdict

Python标准库中的collections模块提供了一个名为defaultdict的强大工具，它非常适合解决这类问题。defaultdict是dict的一个子类，它重写了__missing__方法。当访问一个不存在的键时，defaultdict会自动调用其构造函数中提供的工厂函数来生成一个默认值，并将其赋给该键。

在本例中，我们可以将list作为工厂函数传递给defaultdict。这意味着，每当我们尝试访问一个尚不存在的键时，defaultdict会自动为该键创建一个空列表。

实现步骤

1. 导入defaultdict: 首先从collections模块导入defaultdict。
2. 初始化defaultdict: 创建一个新的defaultdict实例，并将其默认值设置为list。
3. 遍历字典: 遍历所有需要合并的字典。
4. 遍历键值对: 对于每个字典，遍历其所有的键值对。
5. 合并列表值: 使用列表的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(, {'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']。

优点与注意事项

1. Pythonic与简洁: 这种方法利用了defaultdict的特性，使得代码逻辑清晰、简洁，符合Python的优雅风格。
2. 高效性: 相比于手动检查键是否存在并初始化列表，defaultdict在内部进行了优化，通常效率更高。
3. 处理新键: 对于在某些字典中出现而在其他字典中不存在的键，defaultdict能够自动处理，为其创建新的列表并添加值，无需额外的条件判断。
4. 可扩展性: 这种方法可以轻松扩展到合并任意数量的字典。只需将所有字典放入一个列表中进行迭代即可。
5. extend() vs append(): 务必使用extend()方法。如果使用append()，结果将是列表的列表，例如{'a': [['1'], ['2']]}，这不是我们期望的合并列表元素的效果。

总结

当需要在Python中合并多个字典，并且这些字典的键相同且值是列表类型时，使用collections.defaultdict结合列表的extend()方法是最佳实践。这种方法不仅代码简洁、易于理解，而且能够高效地处理各种合并场景，包括新键的自动创建和多个字典的迭代合并。掌握这一技巧将有助于您编写更健壮、更具Pythonic风格的数据处理代码。

以上就是Python中合并具有相同键的字典值列表：高效策略与defaultdict应用的详细内容，更多请关注php中文网其它相关文章！