Question

我有一个字典列表（数据），并希望将其转换为字典（x），如下所示。我正在使用'for loop'来实现。

data = [{'Dept': '0123', 'Name': 'Tom'},
        {'Dept': '0123', 'Name': 'Cheryl'},
        {'Dept': '0123', 'Name': 'Raj'},
        {'Dept': '0999', 'Name': 'Tina'}]
x = {}

for i in data:
    if i['Dept'] in x:
        x[i['Dept']].append(i['Name'])
    else:
        x[i['Dept']] = [i['Name']]

Output:
x -> {'0999': ['Tina'], '0123': ['Tom', 'Cheryl', 'Raj']}

是否有可能以字典理解或任何其他更加pythonic的方式实现上述逻辑？

Answer 1

尽管不是不可能的，但字典理解可能不是最好的选择。我建议您使用defaultdict（https://docs.python.org/2/library/collections.html#collections.defaultdict）：

from collections import defaultdict
dic = defaultdict(list)
for i in data:
    dic[i['Dept']].append(i['Name'])

Answer 2

似乎太复杂了，不允许进入任何重要的代码，但只是为了好玩，在这里你去：

{
    dept: [item['Name'] for item in data if item['Dept'] == dept]
    for dept in {item['Dept'] for item in data}
}

Python Dict理解创建和更新字典

2 个答案: