嵌套字典理解（2 级）答案

【问题标题】：Nested dictionary comprehension (2 level)嵌套字典理解（2 级）
【发布时间】：2021-09-19 03:41:26
【问题描述】：

我有一个字典对象列表

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的方式，意思是可以通过字典理解吗？

【问题讨论】：

标签： python list-comprehension dictionary-comprehension

【解决方案1】：

所以你想玩理解游戏，你想看到一个职业玩家。

忘记 itertools，忘记多个语句。这是一个终极理解，它会让任何编写代码的程序员扔掉键盘，离开房间诅咒你。

data = {parent_id: {d["name"]: d["id"] for d in [i for i in data if i["parent_id"] == parent_id]} for parent_id in set((j["parent_id"] for j in data))}

但实际上，如果要与某人共享，请不要在代码中执行此操作。

【讨论】：

【解决方案2】：

首先，使用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'))
}

【讨论】：