Как сложить элементы списков по другим элементам списков?

Question

[VaniLuksh]

Имеется список списков:
num_list = [[5, 1], [10, 1], [30, 2], [20, 2]]

Нужно сложить первые числа списков по значениям вторых элементов, т.е. сгруппировать по вторым элементам списка

sum_list1 = [15, 1]
sum_list2 = [50, 2]

Comments

[Влад Григорьев]

VaniLuksh, ответ легко гуглиться что бы не мучатся с проверками в листе, используйте словарь, потом если уж так надо преобразуйте в лист

from collections import defaultdict

dan = [[5, 1], [10, 1], [30, 2], [20, 2]]
result = defaultdict(int)

for v, k in dan:
    result[k] += v

print(result)

[VaniLuksh]

Влад Григорьев, вариант рабочий, спасибо)
а про реализацию перебора списка гуглил, но ничего не нашел(

[Влад Григорьев]

VaniLuksh, ну быстрее чем достать по хэшу элемент, вы ничего не найдете.
можете сгрупировать все по ключу, и потом посчитать, но все равно енто будет тот же словарь

Answer 1

[dob]

for e in set([l[-1] for l in num_list]):
    print([e, sum([l[0] for l in num_list if l[1] == e])])

Или получить генератор:

def get_iter(num_list):
    for e in set([l[-1] for l in num_list]):
        yield [sum([l[0] for l in num_list if l[1] == e]), e]

for e in get_iter(num_list):
    print(e)

Comments

[VaniLuksh]

то что надо, спасибо!