Понимание Python Dict для создания и обновления словаря

У меня есть список словарей (данных), и я хочу преобразовать его в словарь (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']}

Можно ли реализовать приведенную выше логику в понимании dict или любым другим более питоническим способом?


python dictionary append dictionary-comprehension defaultdict
person akshat    schedule 16.04.2015    source источник

Ответы (2)


arrow_upward
6
arrow_downward

Понимание dict, хотя и не невозможное, может быть не лучшим выбором. Могу ли я предложить использовать 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'])
person Julien Spronck    schedule 16.04.2015

arrow_upward
5
arrow_downward

Это кажется слишком сложным, чтобы его можно было допустить в любой код, который имеет хоть какое-то значение, но просто для удовольствия, вот, пожалуйста:

{
    dept: [item['Name'] for item in data if item['Dept'] == dept]
    for dept in {item['Dept'] for item in data}
}
person Underyx    schedule 16.04.2015
comment
Я вообще не считаю это слишком сложным. Однако это медленнее в смысле большого O, потому что данные необходимо повторно сканировать один раз для каждого найденного уникального ключа - в худшем случае O (N ^ 2). - person Karl Knechtel; 16.04.2015
comment
Спасибо Underyx, это отвечает на мой вопрос и отлично работает. И да, нужно время, чтобы понять. :). Поэтому я выберу то, что предложил @Julien Spronck. - person akshat; 16.04.2015