Как объединить различные значения одного ключа, а затем суммировать совпадающие значения другого ключа?
Я сделал цикл, который выдает мне данные в следующем формате:
name_quant = [{'name_id': 'S00004', 'quantity': '1'}, {'name_id': 'S00004', 'quantity': '2'}, {'name_id': 'S00003', 'quantity': '1'},
{'name_id': 'S00003', 'quantity': '2'}, {'name_id': 'S00003', 'quantity': '2'}, {'name_id': 'S00002', 'quantity': '1'}]
Я использовал следующий цикл для получения вышеуказанных значений:
namesequence = EventSequence.objects.filter(description="names").values("Details")
name_quant = [{ 'name_id': e['element'][33:39],
'quantity': e['element'][50:51] } for e in namesequence ]
Итак, мой вопрос в том, как я могу объединить name_ids и суммировать количество совпадающих name_ids так, чтобы получить результат, подобный этому:
name_sum = [{'name_id': 'S00001', 'quantity': '160'}, {'name_id': 'S00002', 'quantity': '50'}, {'name_id': 'S00003', 'quantity': '40'}, {'name_id': 'S00004', 'quantity': '90'}]
Я бы использовал функцию sum в Django, но мне нужно сначала сделать подзапись и зациклить значение, что немного усложняет задачу :/
Любая помощь приветствуется!
Если я правильно понимаю вопрос, похоже, что требование заключается в консолидации ключей (имя_имя) по количеству. Я не могу понять, как требуемые выходные значения получаются из входных данных образца, но это может быть потому, что они неполные.
name_quant = [{'name_id': 'S00004', 'quantity': '1'}, {'name_id': 'S00004', 'quantity': '2'}, {'name_id': 'S00003', 'quantity': '1'},
{'name_id': 'S00003', 'quantity': '2'}, {'name_id': 'S00003', 'quantity': '2'}, {'name_id': 'S00002', 'quantity': '1'}]
td = dict()
for e in name_quant:
nid = e['name_id']
td[nid] = td.get(nid, 0) + int(e['quantity'])
new_list = [{'name_id': k, 'quantity': str(v)} for k, v in td.items()]
print(new_list)
Выход:
[{'name_id': 'S00004', 'quantity': '3'}, {'name_id': 'S00003', 'quantity': '5'}, {'name_id': 'S00002', 'quantity': '1'}]
Если список name_quant большой, я предпочитаю использовать pandas для выполнения groupby staff:
import pandas as pd
name_quant = [{'name_id': 'S00004', 'quantity': '1'}, {'name_id': 'S00004', 'quantity': '2'},
{'name_id': 'S00003', 'quantity': '1'},
{'name_id': 'S00003', 'quantity': '2'}, {'name_id': 'S00003', 'quantity': '2'},
{'name_id': 'S00002', 'quantity': '1'}]
df = pd.DataFrame.from_records(name_quant)
df['quantity'] = df['quantity'].astype(int)
results = df.groupby(['name_id']).agg({'quantity': 'sum'}).to_records() # [('S00002', 1) ('S00003', 5) ('S00004', 3)]
grouped_name_quant = [{'name_id': x[0], 'quantity': x[1]} for x in results]
print(grouped_name_quant)
Вывод:
[{'name_id': 'S00002', 'quantity': 1}, {'name_id': 'S00003', 'quantity': 5}, {'name_id': 'S00004', 'quantity': 3}]