Как сложить значения у одинаковых ключей в списке?

Question

[organica]

Господа, помогите пожалуйста советом: имеются следующие вводные:
Есть список, в котором лежат еще три списка, возвращаемые функциями

united_scores = [self.find_common_groups(), self.find_common_friends(), self.compare_and_set_score()]

Элементы вложенных списков - словари в следующем формате:

{'id': 359631473, 'score': 6}
{'id': 133979437, 'score': 2}
{'id': 53245825, 'score': 2}
{'id': 225959252, 'score': 2}

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

def unite_lists(self):
        united_scores = [self.find_common_groups(), self.find_common_friends(), self.compare_and_set_score()]
        for element in united_scores:
            result = dict(functools.reduce(operator.add,
                                           map(collections.Counter, element)))
            print(result)

В Результате получаю сумму всех id и их score по трем входящим спискам...:

{'id': 47308545803, 'score': 2555}
{'id': 23604130033, 'score': 1340}
{'id': 12414787583, 'score': 946}

Как наиболее элегантно решить мою задачу?

Answer 1

[Дмитрий Темников]

Используйте пакет Pandas . Ваша задача решается в одну строку

import pandas as pd

lst=[{'id': 2785280, 'score': 13},
 {'id': 66560002, 'score': 4},
 {'id': 180432899, 'score': 7},
 {'id': 52793348, 'score': 5},
 {'id': 86411270, 'score': 5},
 {'id': 2334728, 'score': 2},
 {'id': 2523146, 'score': 4},
 {'id': 159318028, 'score': 5},
]

df=pd.DataFrame(lst)
df=df.groupby('id')['score'].sum()

Answer 2

[0xD34F]

flat = [ n for m in united_scores for n in m ]
ids = set(n['id'] for n in flat)
grouped = [ { 'id': n, 'score': sum(m['score'] for m in flat if m['id'] == n) } for n in ids ]

Comments

[organica]

Респект!

[organica]

А можно еще вопрос попутно?
Как отфильтровать эти элементы по убыванию? Фильтрацию по значению ключей внутри словаря я знаю, а как это проделать в списке, включающим в себя словари?

def unite_lists(self):
        united_scores = [self.find_common_groups(), self.find_common_friends(), self.compare_and_set_score()]
        flat = [n for m in united_scores for n in m]
        ids = set(n['id'] for n in flat)
        grouped = [{'id': n, 'score': sum(m['score'] for m in flat if m['id'] == n)} for n in ids]
        pprint(grouped)

Результат

[{'id': 2785280, 'score': 13},
 {'id': 66560002, 'score': 4},
 {'id': 180432899, 'score': 7},
 {'id': 52793348, 'score': 5},
 {'id': 86411270, 'score': 5},
 {'id': 2334728, 'score': 2},
 {'id': 2523146, 'score': 4},
 {'id': 159318028, 'score': 5},
.....
]

[0xD34F]

organica,

Как отфильтровать эти элементы по убыванию?

Никак. То, что вы сейчас спросили - бессмысленная ахинея.

Наверное, вместо "отфильтровать" должно было быть "отсортировать". Сортируйте набор id:

ids = sorted(set(n['id'] for n in flat), reverse=True)

[organica]

0xD34F, Спасибо большое за совет, но сделал свою незамысловатую конструкцию для выполнения задачи. Очень тяжело даются для понимания генераторы словарей и списков, которые вы предлагаете. Надо разбираться сидеть с этой темой ...

def unite_lists(self):
        list_of_winners = []
        united_scores = [self.find_common_groups(), self.find_common_friends(), self.compare_and_set_score()]
        flat = [n for m in united_scores for n in m]
        ids = set(n['id'] for n in flat)
        grouped = [{'id': n, 'score': sum(m['score'] for m in flat if m['id'] == n)} for n in ids]
        top_scores = sorted(set(n['score'] for n in flat), reverse=True)
        for el in top_scores[0:4]:
            for item in grouped:
                if item['score'] == el:
                    if len(list_of_winners) < 5:
                        list_of_winners.append(item)
        pprint(list_of_winners)

Результат

[{'id': 3973347, 'score': 16},
 {'id': 1621076, 'score': 16},
 {'id': 4636997, 'score': 12},
 {'id': 4551567, 'score': 12},
 {'id': 13331707, 'score': 12}]

[o5a]

organica, Для сортировки массива по необычному условию можно использовать параметр key, в который передается функция ранжирования. В данном случае это можно сделать так

grouped = # массив суммированных значений из предыдущего кода

grouped.sort(key=lambda x: x['score'], reverse=True)

или дополнить следующим образом, чтобы сортировал по id в случае одинаковых score

grouped.sort(key=lambda x: (x['score'], x['id']), reverse=True)