使用Python减少对列表

时间:2012-10-21 16:42:23

标签: python map functional-programming reduce accumulator

我正在尝试将列表中的一堆元素组合在一起以创建最终对象,其方式类似于创建对象的总和。我正在尝试在reduce上使用一个简单的变体,您可以在其中考虑对列表而不是平面列表来执行此操作。我想按照以下方式做点什么:

nums = [1, 2, 3]
reduce(lambda x, y: x + y, nums)

除了我想在数字列表nums中的每个元素的特定总和中添加其他信息。例如,对于列表中的每对(a,b),将总和运行为(a + b):

nums = [(1, 0), (2, 5), (3, 10)]
reduce(lambda x, y: (x[0]+x[1]) + (y[0]+y[1]), nums)

这不起作用:

>>> reduce(lambda x, y: (x[0]+x[1]) + (y[0]+y[1]), nums)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <lambda>
TypeError: 'int' object is unsubscriptable

为什么不起作用?我知道我可以将nums编码为平面列表 - 这不是重点 - 我只是希望能够创建一个可以迭代一对对的reduce操作,或者同时迭代两个相同长度的列表和两个列表中的池信息。感谢。

1 个答案:

答案 0 :(得分:6)

查看传递给reduce的lambda:

f = lambda x, y: (x[0]+x[1]) + (y[0]+y[1])

f返回的值将作为参数传递给f的另一个调用。但是,虽然f期望其参数是成对的,但它返回的值是int。你需要让这个函数返回一对。例如,这将分别总结左侧和右侧:

>>> nums = [(1, 0), (2, 5), (3, 10)]
>>> reduce(lambda x, y: (x[0] + y[0], x[1] + y[1]), nums)
(6, 15)

您可以做的另一件事是以不同于列表元素的方式处理累积值:您可以将累积值设为int,而列表元素为对。如果这样做,您必须将initializer参数传递给reduce,以便将累加器正确初始化为int

>>> nums = [(1, 0), (2, 5), (3, 10)]
>>> reduce(lambda acc, y: acc + y[0] + y[1], nums, 0)
21