如何展平嵌套的python字典?

2024-10-03 11:25:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我正试着把嵌套字典弄平:

dict1 = {
    'Bob': {
        'shepherd': [4, 6, 3],
        'collie': [23, 3, 45],
        'poodle': [2, 0, 6],
    },
    'Sarah': {
        'shepherd': [1, 2, 3],
        'collie': [3, 31, 4],
        'poodle': [21, 5, 6],
    },
    'Ann': {
        'shepherd': [4, 6, 3],
        'collie': [23, 3, 45],
        'poodle': [2, 10, 8],
    }
}

我想显示列表中的所有值: [4,6,3,23,3,45,2,0,6,1,2,3,…,2,10,8]

我的第一个想法是这样做:

dict_flatted = [ i for name in names.values() for dog in dogs.values() for i in dog]

尽管我犯了错误。我很乐意给你一些处理方法。


Tags: in列表for字典dictbobvaluesdog
3条回答

你试图使用不存在的变量。

用这个

dict_flatted = [ i for names in dict1.values() for dog in names.values() for i in dog]

您可以使用一个简单的递归函数,如下所示。

def flatten(d):    
    res = []  # Result list
    if isinstance(d, dict):
        for key, val in d.items():
            res.extend(flatten(val))
    elif isinstance(d, list):
        res = d        
    else:
        raise TypeError("Undefined type for flatten: %s"%type(d))

    return res


dict1 = {
    'Bob': {
        'shepherd': [4, 6, 3],
        'collie': [23, 3, 45],
        'poodle': [2, 0, 6],
    },
    'Sarah': {
        'shepherd': [1, 2, 3],
        'collie': [3, 31, 4],
        'poodle': [21, 5, 6],
    },
    'Ann': {
        'shepherd': [4, 6, 3],
        'collie': [23, 3, 45],
        'poodle': [2, 10, 8],
    }
}

print( flatten(dict1) )

递归函数可以是单行:

def flatten(d):
    return d if isinstance(d, list) else sum((flatten(i) for i in d.values()), [])

Nota:是一个duplicate question的答案

相关问题 更多 >