使用 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行代码，还不显摆技巧。在 virtuoso 的情况下，只需要 3 行代码。

为了快速扁平化这个嵌套字典，我们需要处理来自 downnet 的字段。例如，对于路径 b->e->f->4，我们首先将最里面的 {'f': 4} 转换为一个元组 ('f', 4)。然后，向上抛出这个元组，得到元组('e', ('f', 4))。我们在f前面拼接e成为：('e_f', 4)，继续向上抛得到('b', ('e_f', 4))。然后将 b 连接到 e_f 得到 ('b_e_f', 4)。完成一条线的组装。

这个逻辑如果使用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 关键字，字典的键会由内而外逐层组装，形成一条完整的路径，就像流水线一样。