Python使用yield压平嵌套字典的超简单方法

嵌套字典压平需求与yield方案

在处理数据时，我们常会遇到字典内部嵌套其他字典的情况，例如：

nest_dict = {
  'a': 1,
  'b': {
    'c': 2,
    'd': 3,
    'e': {'f': 4}
  },
  'g': {'h': 5},
  'i': 6,
  'j': {'k': {'l': {'m': 8}}}
}

是否存在一种简便方式，将其转换为扁平结构，结果如下：

{
  'a': 1,
  'b_c': 2,
  'b_d': 3,
  'b_e_f': 4,
  'g_h': 5,
  'i': 6,
  'j_k_l_m': 8
}

面对此类问题，递归是常见的解决思路，但实现起来往往需要编写较多的代码行。

本文将演示如何使用yield关键字来完成这一任务。在不追求代码极简的情况下，仅需8行代码即可实现；若追求极致精简，甚至只需3行代码。

从内向外组装路径的逻辑解析

要实现嵌套字典的快速压平，处理逻辑应从内层向外层推进。以路径b->e->f->4为例，首先将最内层的{'f': 4}转换为元组('f', 4)。随后，将该元组向上层传递，得到元组('e', ('f', 4))。接着，将键e拼接到前一个键f之前，形成新的键名('e_f')，并与值4组合为('e_f', 4)。继续向上层传递，得到('b', ('e_f', 4))，最后将键b拼接到e_f之前，最终生成('b_e_f', 4)，从而完成一条完整路径的组装。

yield关键字实现代码与效果

若使用yield关键字实现上述逻辑，代码结构如下：

def flat(x):
  for key, value in x.items():
    if isinstance(value, dict):
      for k, v in flat(value):
        k = f'{key}_{k}'
        yield (k, v)
    else:
      yield (key, value)

程序的运行结果如图所示：

通过引入yield关键字，字典的键值对在遍历过程中如同流水线作业，逐层从内向外组装，最终生成完整的路径键名。

在后续文章中，我们将继续探讨如何利用yield关键字处理字典与列表混合嵌套的复杂情况。

总结

以上内容涵盖了本文的核心知识点，希望能为您的学习或工作提供有价值的参考，感谢支持。