按索引在列表值之间求和

Question

我正在尝试按索引汇总一堆列表，例如：

a = [[1,2,3], [2,3,4], [3,4,5]]

output = [ reduce(lambda x, y: x[_count] + y[_count], a) 
           if _count in [0,2] else "" for _count in xrange(len(a)) ]

它一定适合我，预期的输出是：

>>> output
[6, "", 12]

但我得到了：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <lambda>
TypeError: 'int' object has no attribute '__getitem__'

可能我不了解reduce的引擎。

我做错了吗？

Answer 1

我不明白你为什么使用reduce。 reduce 也适用于列表，并且由于它是整数列表，因此 lambda 表达式中的至少 y 将是一个不可编写脚本的 integer（例如，您无法获得 2[1]）。

但是，您可以为此使用 sum 和 列表理解：

output = [sum(ai) if i in (0,2) else "" for i,ai in enumerate(a)]

如果您想对 转置 求和（因此两者不等价），您还可以使用生成器而不是总和：

output = [sum(ai[i] for ai in a) if i in (0,2) else "" for i in range(len(a))]

或者如果你想计算a[0][0]+a[1][0]+a[2][0]+...等，你可以使用reduce，累加器应该以0开头并且你将y[_count]添加到x（不是x[_count]）：

from functools import reduce

output = [reduce(lambda x,y:x+y[i],a,0) if i in (0,2) else "" for i in range(len(a))]