码迷,mamicode.com
首页 > 其他好文 > 详细

使用 yield 压平嵌套字典有多简单?

时间:2020-12-07 12:49:42      阅读:14      评论:0      收藏:0      [点我收藏+]

标签:多层   就是   产品   需要   png   简单的   使用   实现   ict   

使用 yield 压平嵌套字典有多简单?

技术图片

摄影:产品经理
买单:kingname
我们经常遇到各种字典套字典的数据,例如:


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关键字来解决字典与列表混合嵌套的情况。

推荐阅读:一日一技:如何把多层嵌套的列表展平

技术图片

使用 yield 压平嵌套字典有多简单?

标签:多层   就是   产品   需要   png   简单的   使用   实现   ict   

原文地址:https://blog.51cto.com/15023263/2558925

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!