0

0

使用 ChainMap 实现 Python 字典的深度合并

心靈之曲

心靈之曲

发布时间:2025-08-11 18:16:41

|

355人浏览过

|

来源于php中文网

原创

使用 chainmap 实现 python 字典的深度合并

本文深入探讨了如何利用 Python 的 collections.ChainMap 实现复杂字典的深度合并。针对 ChainMap 默认的浅层合并行为无法满足嵌套字典合并的需求,文章提出了一种自定义 DeepChainMap 类的方法。通过重写 __getitem__ 方法,该方案能够递归地合并具有相同键的嵌套字典,从而实现真正的深度合并,同时保留 ChainMap 的惰性查找特性。

理解 ChainMap 的基本行为与局限性

collections.ChainMap 是 Python 标准库中一个非常实用的工具,它将多个字典或映射组合成一个单一的、可更新的视图。当查找一个键时,ChainMap 会按照传入字典的顺序依次查找,返回第一个找到的键对应的值。这在需要按优先级合并配置或环境变量等场景下非常有用。

考虑以下两个字典 a 和 b:

a = {'123': {'player': 1,
             'opponent': 2},
     '18': {'player': 10,
            'opponent': 12}
    }

b = {'123': {'winner': 1},
     '180': {'winner': 2}
    }

如果直接使用 ChainMap(a, b) 并尝试将其转换为字典,我们会发现 ChainMap 默认的行为是浅层合并。对于相同的顶级键,它会优先使用第一个字典中的值,而不会尝试合并嵌套的字典:

from collections import ChainMap

print(dict(ChainMap(a, b)))
# 输出: {'123': {'player': 1, 'opponent': 2}, '180': {'winner': 2}, '18': {'player': 10, 'opponent': 12}}

可以看到,键 '123' 的值直接取自 a,b 中 '123' 下的 {'winner': 1} 被完全忽略了。这与我们期望的深度合并结果,即 '123' 下的字典应合并为 {'player': 1, 'opponent': 2, 'winner': 1},相去甚远。

立即学习Python免费学习笔记(深入)”;

实现深度合并:自定义 DeepChainMap

要实现对嵌套字典的深度合并,我们需要扩展 ChainMap 的行为,使其在遇到嵌套字典时能够递归地进行合并。这可以通过重写 DeepChainMap 类的 __getitem__ 方法来实现。当 ChainMap 尝试获取一个键的值时,如果该值本身是一个字典,我们则递归地为这个嵌套字典创建另一个 DeepChainMap 实例,从而实现深度合并。

Skybox AI
Skybox AI

一键将涂鸦转为360°无缝环境贴图的AI神器

下载

以下是 DeepChainMap 的实现:

from collections import ChainMap

class DeepChainMap(ChainMap):
    """
    ChainMap 的变体,支持对嵌套字典的深度合并。
    当获取一个键的值时,如果该值是字典,则递归地合并所有映射中该键对应的字典。
    """
    def __getitem__(self, key):
        # 收集所有映射中存在给定键的值
        values = (mapping[key] for mapping in self.maps if key in mapping)

        try:
            # 获取第一个找到的值
            first = next(values)
        except StopIteration:
            # 如果所有映射中都不存在该键,则调用父类的 __missing__ 方法(通常会抛出 KeyError)
            return self.__missing__(key)

        # 如果第一个值是字典,则递归地为所有找到的字典值创建 DeepChainMap
        if isinstance(first, dict):
            # 注意:这里将 first 放在第一个,以确保其内容优先,
            # 并且后续的 values 会按 ChainMap 的顺序进行合并
            return self.__class__(first, *values)

        # 如果不是字典,直接返回第一个找到的值
        return first

    def __repr__(self):
        # 为了更清晰的表示,将 DeepChainMap 转换为普通字典的字符串表示
        return repr(dict(self))

__getitem__ 方法解析

  1. values = (mapping[key] for mapping in self.maps if key in mapping): 这一行是核心。它遍历 DeepChainMap 内部存储的所有映射(即传入的原始字典),并生成一个迭代器,其中包含所有找到的、与给定 key 关联的值。这样,即使 key 在多个字典中都存在,我们也能获取到它们所有的值。
  2. first = next(values): 尝试从 values 迭代器中获取第一个值。这个值是根据 ChainMap 的查找顺序(即传入字典的顺序)找到的第一个值。
  3. except StopIteration: return self.__missing__(key): 如果 values 迭代器为空(即在任何映射中都没有找到 key),则抛出 StopIteration 异常。我们捕获这个异常并调用 self.__missing__(key),这通常会导致 KeyError,符合字典找不到键的行为。
  4. *`if isinstance(first, dict): return self.class(first, values)**: 这是实现深度合并的关键。如果first值是一个字典,这意味着我们遇到了一个需要深度合并的嵌套结构。此时,我们递归地创建一个新的DeepChainMap实例。这个新的DeepChainMap将以first字典作为其第一个映射,然后将values迭代器中剩余的所有字典(即在其他映射中找到的、与相同key关联的字典)作为后续映射传入。这样,新的DeepChainMap` 会负责合并这些嵌套字典。
  5. return first: 如果 first 值不是字典(例如,它是字符串、数字、列表等),则不需要深度合并,直接返回 first 即可。

__repr__ 方法解析

__repr__ 方法被重写是为了提供一个更友好的字符串表示。默认情况下,ChainMap 的 repr 会显示其内部的 maps 列表,这对于调试和理解 DeepChainMap 的最终状态不太直观。通过将其转换为 dict(self) 再进行 repr,我们可以直接看到合并后的字典结构,就像一个普通的字典一样。

使用 DeepChainMap 进行深度合并

现在,我们可以使用自定义的 DeepChainMap 来实现我们最初的目标:

# 重新定义原始字典
a = {'123': {'player': 1,
             'opponent': 2},
     '18': {'player': 10,
            'opponent': 12}
    }

b = {'123': {'winner': 1},
     '180': {'winner': 2}
    }

# 使用 DeepChainMap 进行合并
merged_map = DeepChainMap(a, b)
print(merged_map)
# 预期输出: {'123': {'winner': 1, 'player': 1, 'opponent': 2}, '180': {'winner': 2}, '18': {'player': 10, 'opponent': 12}}

# 转换为普通字典以进行进一步操作(如果需要)
final_dict = dict(merged_map)
print(final_dict)
# 输出: {'123': {'winner': 1, 'player': 1, 'opponent': 2}, '180': {'winner': 2}, '18': {'player': 10, 'opponent': 12}}

从输出可以看出,对于键 '123',嵌套字典 { 'player': 1, 'opponent': 2 } 和 { 'winner': 1 } 已经被成功合并为一个完整的字典 { 'player': 1, 'opponent': 2, 'winner': 1 }。同时,非重叠的键 '18' 和 '180' 也被正确保留。

注意事项与总结

  • 惰性查找特性: DeepChainMap 继承了 ChainMap 的惰性查找特性。这意味着只有当实际访问某个键时,才会执行查找和潜在的深度合并操作。这对于处理非常大的字典集合,但只访问其中一部分数据的情况,可以提高效率。
  • 合并顺序: 合并的优先级仍然遵循 ChainMap 的规则,即传入 DeepChainMap 的字典顺序。对于非字典类型的值,第一个找到的值会被返回。对于字典类型的值,它们会递归地合并,但后续字典中的同名键会覆盖前面字典中的同名键(这是 dict.update() 或字典合并的常见行为)。
  • 适用场景: DeepChainMap 特别适用于需要将多个配置源(如默认配置、用户配置、命令行参数)进行深度合并的场景,尤其当这些配置中包含嵌套结构时。
  • 与递归更新的区别: 如果你仅仅需要一个最终的、完全合并后的新字典,并且不关心 ChainMap 的惰性查找特性,那么编写一个递归函数来更新字典可能是另一种更直接的方式。然而,DeepChainMap 提供了一种优雅且面向对象的方式来处理这类深度合并问题,尤其是在需要维护多个字典视图的场景下。

通过自定义 DeepChainMap,我们成功扩展了 collections.ChainMap 的功能,使其能够处理更复杂的深度合并需求,为 Python 开发者提供了更强大的字典操作工具。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

778

2023.08.22

go语言 面向对象
go语言 面向对象

本专题整合了go语言面向对象相关内容,阅读专题下面的文章了解更多详细内容。

56

2025.09.05

java面向对象
java面向对象

本专题整合了java面向对象相关内容,阅读专题下面的文章了解更多详细内容。

52

2025.11.27

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

299

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

212

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1502

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

624

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

633

2024.03.22

C++ 设计模式与软件架构
C++ 设计模式与软件架构

本专题深入讲解 C++ 中的常见设计模式与架构优化,包括单例模式、工厂模式、观察者模式、策略模式、命令模式等,结合实际案例展示如何在 C++ 项目中应用这些模式提升代码可维护性与扩展性。通过案例分析,帮助开发者掌握 如何运用设计模式构建高质量的软件架构,提升系统的灵活性与可扩展性。

0

2026.01.30

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.7万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号