使用 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)，继续往上抛出，获得('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关键字，字典的key会像是在流水线上同样，一层一层从内向外进行组装，从而造成完整的路径。

在下一篇文章中，咱们继续使用yield关键字来解决字典与列表混合嵌套的状况。