如果相同的值,则为元组

时间:2013-04-11 16:01:08

标签: python list tuples

这是我的元组列表:

regions = [(23.4, 12, 12341234), 
           (342.23, 19, 12341234), 
           (4312.3, 12, 12551234), 
           (234.2, 12, 12341234)]

我试图在元组列表中对第一个索引值求和,其中索引1和2的值相同。请注意,区域[0]和区域[3]在索引1和2处具有相同的值。

我想要的清单是:

result = [(257.6, 12, 12341234), 
          (342.23, 19, 12341234), 
          (4312.3, 12, 12551234)]

我意识到我可能需要先将它保存为字典,可能将第二个值作为第一个键,将第三个值作为第二个键并将其汇总(如果已存在)。只是想知道是否有更好或更简单的方法:我想也许使用一些交叉功能。

1 个答案:

答案 0 :(得分:6)

from collections import defaultdict

sums = defaultdict(float)
for c, a, b in regions:
    sums[a, b] += c
result = [(csum, a, b) for (a, b), csum in sums.iteritems()]

没有内置功能可以做到这一点;这太专业了。