时间戳

首页 > 解决方案 > 嵌套字典理解（2 级）

python - 嵌套字典理解（2 级）

问题描述

我有一个字典对象列表

data = [
    {
        'id': 1,
        'parent_id': 101 ,
        'name': 'A'        
    },
    {
        'id': 2,
        'parent_id': 101,
        'name': 'B'        
    },
    {
        'id': 3,
        'parent_id': 102,
        'name': 'C'        
    },
    {
        'id': 4,
        'parent_id': 102,
        'name': 'D'        
    }
]

我想将其转换为嵌套字典。

结构：

{
    parent_id_value: {
        name_value: id_value
    }
}

字典对象的样本列表的结果应该是这样的

{
    101: {
        'A': 1,
        'B': 2
    },
    102: {
        'C': 3,
        'D': 4
    }
}

我知道我们可以运行一个for循环并使用setdefault来设置/获取paren_id值，然后将名称和 ID 添加为键、值

new_dic = {}
for i in data:
    new_dic.setdefault(i['parent_id'], {})[i['name']] = i['id']
print(new_dic)

但我正在寻找一种更pythonic的方式，意思是可以通过字典理解吗？

标签： pythonlist-comprehensiondictionary-comprehension

解决方案

operator.itemgetter首先，使用提供键对列表进行排序：

data.sort(key=itemgetter('parent_id'))

然后itertools.groupby用于在嵌套的 dict-comprehension 中对预期的部分进行分组：

data = {
    key: {item['name']: item['id'] for item in group}
        for key, group in groupby(data, itemgetter('parent_id'))
}

我不建议写一个单行，但你可以这样做sorted：

data = {
    key: {item['name']: item['id'] for item in group}
        for key, group in groupby(sorted(data, key=itemgetter('parent_id')), itemgetter('parent_id'))
}

推荐阅读