Question

我有一个字典对象列表

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

Answer 1

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

Answer 2

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

忘记itertools，忘记多条语句。这是最终的理解，它会让任何在你的代码上工作的程序员扔掉键盘，离开房间诅咒你。

<Installation>
   <InstallationTarget Id="Microsoft.VisualStudio.Pro" Version="[16.3.29209,17.0)">
      <ProductArchitecture>x86</ProductArchitecture>
   </InstallationTarget>
   <InstallationTarget Id="Microsoft.VisualStudio.Pro" Version="[17.0,18.0)">
      <ProductArchitecture>amd64</ProductArchitecture>
   </InstallationTarget>
</Installation>

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

嵌套字典理解（2级）

2 个答案: