33

我有一个类似于此的元组列表:

l = [(1, 2), (3, 4), (5, 6), (7, 8), (9, 0)]

我想创建一个简单的单线,它会给我以下结果:

r = (25, 20) or r = [25, 20] # don't care if tuple or list.

这就像执行以下操作:

r = [0, 0]
for t in l:
  r[0]+=t[0]
  r[1]+=t[1]

我确信这是非常简单的事情,但我想不出来。

注意:我已经看过类似的问题:

如何对元组内的一组列表中的第一个值求和?

如何在 Python 的元组列表中对每个元组中的第一个值求和?

4

3 回答 3

65

使用zip()sum()

In [1]: l = [(1, 2), (3, 4), (5, 6), (7, 8), (9, 0)]

In [2]: [sum(x) for x in zip(*l)]
Out[2]: [25, 20]

或者:

In [4]: map(sum, zip(*l))
Out[4]: [25, 20]

timeit结果:

In [16]: l = [(1, 2), (3, 4), (5, 6), (7, 8), (9, 0)]*1000

In [17]: %timeit [sum(x) for x in zip(*l)]
1000 loops, best of 3: 1.46 ms per loop

In [18]: %timeit [sum(x) for x in izip(*l)]       #prefer itertools.izip
1000 loops, best of 3: 1.28 ms per loop

In [19]: %timeit map(sum, zip(*l))
100 loops, best of 3: 1.48 ms per loop

In [20]: %timeit map(sum, izip(*l))                #prefer itertools.izip
1000 loops, best of 3: 1.29 ms per loop
于 2013-01-06T09:33:53.480 回答
3

我想在给定的答案中添加一些内容:

如果我有一个dict数组,例如

l = [{'quantity': 10, 'price': 5},{'quantity': 6, 'price': 15},{'quantity': 2, 'price': 3},{'quantity': 100, 'price': 2}]

我想获得两个(或更多)计算数量的总和,例如数量和价格*数量的总和

我可以:

(total_quantity, total_price) = (
sum(x) for x in zip(*((item['quantity'],
                       item['price'] * item['quantity'])
                      for item in l)))

代替:

total_quantity = 0
total_price = 0
for item in l:
     total_quantity += item['quantity']
     total_price += item['price'] * item['quantity']

也许第一个解决方案可读性较差,但更“pythonesque”:)

于 2015-09-24T16:16:57.073 回答
3

不使用 zip

sum(e[0] for e in l), sum(e[1] for e in l)
于 2017-07-07T23:03:25.157 回答