Soma dos valores em lista quando condição for atendida

votos
1

Eu tenho uma lista python lcontendo instâncias da classe Element:

class Element:
    def __init__(self, id, value):
        self.id = id
        self.value = value

l = [Element(1, 100), Element(1, 200), Element(2, 1), Element(3, 4), Element(3, 4)]

Agora eu quero somar todos valueos membros das classes Elementsse o seu idé igual para obter esta lista:

l = [Element(1, 300), Element(2, 1), Element(3, 8)]

Qual é a maneira mais Python de fazer isso?

Publicado 19/12/2018 em 14:11
fonte usuário
Em outras línguas...                            


3 respostas

votos
6

Há (quase?) Nada que itertoolsnão possa fazer. Dê uma olhada groupby:

from itertools import groupby
from operator import attrgetter


class Element:
    def __init__(self, id, value):
        self.id = id
        self.value = value
    def __repr__(self):  # kudos @mesejo
        return "Element({}, {})".format(self.id, self.value)

l = [Element(1, 100), Element(1, 200), Element(2, 1), Element(3, 4), Element(3, 4)]

l.sort(key=attrgetter('id'))  # if it is already sorted by 'id', comment-out

res = [Element(g, sum(sub.value for sub in k)) for g, k in groupby(l, key=attrgetter('id'))]

o que resulta em:

print(res)   # [Element(1, 300), Element(2, 1), Element(3, 8)]
Respondeu 19/12/2018 em 14:17
fonte usuário

votos
2

Uma forma seria a de criar um defaultdictque mapeia IDs de somas de valores. Então podemos tomar esses resultados e usá-los para construir uma nova lista de Elements. Uma maneira de fazer isso é usar starmappara mapear os itens desse dicionário para os argumentos paraElement

from collections import defaultdict
from itertools import starmap

class Element:
    def __init__(self, id, value):
        self.id = id
        self.value = value
    def __repr__(self):
        return "Element({}, {})".format(self.id, self.value)

l = [Element(1, 100), Element(1, 200), Element(2, 1), Element(3, 4), Element(3, 4)]

d = defaultdict(int)

for e in l:
    d[e.id] += e.value

print(list(starmap(Element, d.items())))
# [Element(1, 300), Element(2, 1), Element(3, 8)]
Respondeu 19/12/2018 em 14:18
fonte usuário

votos
1

Você também pode obter o resultado desejado usando setpara obter apenas os ids únicos e sumpara totalizar os valores. Por exemplo:

class Element:
    def __init__(self, id, value):
        self.id = id
        self.value = value

l = [Element(1, 100), Element(1, 200), Element(2, 1), Element(3, 4), Element(3, 4)]

ids = set(elem.id for elem in l)
totals = [Element(i, sum(elem.value for elem in l if elem.id == i)) for i in ids]
# [Element(1, 300), Element(2, 1), Element(3, 8)]
Respondeu 19/12/2018 em 14:36
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more