使用Python字典推导式创建和更新字典

Question

我有一个字典列表（data），想将其转换为以下的字典（x）。我使用以下的“for循环”来实现。

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']}

“是否可以在字典推导或其他更具有Python风格的方式中实现上述逻辑？”

- akshat

Answer 1

虽然使用字典推导不是不可能，但可能不是最好的选择。我可以建议使用 defaultdict：

from collections import defaultdict

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

- Julien Spronck

Answer 2

看起来过于复杂，不适合用于任何有丝毫重要性的代码中，但只是为了好玩，这里是：

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

- Underyx

---

我并不认为这太复杂了。然而，从大O的角度来看，它比较慢，因为数据必须为每个找到的唯一键重新扫描一次 - 最坏情况下是O(N^2)。 - Karl Knechtel