首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >满足条件时列表中的值之和

满足条件时列表中的值之和
EN

Stack Overflow用户
提问于 2018-12-19 22:11:48
回答 3查看 94关注 0票数 1

我有一个包含类Element实例的python列表l

代码语言:javascript
复制
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)]

现在我想对类Elements的所有value成员求和,如果它们的id等于获得这个列表:

代码语言:javascript
复制
l = [Element(1, 300), Element(2, 1), Element(3, 8)]

要做到这一点,最具蟒蛇色彩的方式是什么?

EN

Stack Overflow用户

回答已采纳

发布于 2018-12-19 22:17:18

有(差不多?)没有itertools做不到的事情。看一看groupby

代码语言:javascript
复制
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'))]

这会导致:

代码语言:javascript
复制
print(res)   # [Element(1, 300), Element(2, 1), Element(3, 8)]
票数 6
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/53853038

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档