我可以在reduce函数中添加一个迭代器吗？答案

【问题标题】：Can I add an iterator to the reduce function?我可以在reduce函数中添加一个迭代器吗？
【发布时间】：2021-05-23 16:06:18
【问题描述】：

假设我有以下列表：A = [1,2,3,4]，通过使用reduce 函数，找到元素的乘积，我可以做到

prodA = reduce(lambda x,y: x*y, A)

但是，如果我有另一个列表B=[9,8,7,6]，我是否仍然可以使用reduce 函数来执行以下操作？（从上到下表示步骤）

9
(9+1)* 2
((9+1) *2)+8
(((9+1) *2)+8)*3
((((9+1) *2)+8)*3)+7
(((((9+1) *2)+8)*3)+7)*4
((((((9+1) *2)+8)*3)+7)*4)+6

我不太确定是否可以将列表B 的迭代器添加到reduce 函数中。我怎么能这样做？非常感谢！

【问题讨论】：

【解决方案1】：

这看起来像是zip 的工作。具体来说，我们要将两个列表压缩在一起，然后我们会将归约函数表示为一个接受元组而不是简单整数的函数。

将我们的列表压缩在一起可以得到我们

>>> list(zip(A, B))
[(1, 9), (2, 8), (3, 7), (4, 6)]

您的函数在每一步都乘以 A 的元素，然后添加 B 的元素。因此，从1 开始（这是一个合理的默认值，因为我们要做的第一件事是相乘，所以1 将是第一个操作的标识），乘以元组的第一个元素并加上第二个元素。

reduce(lambda x, y: x * y[0] + y[1], zip(A, B), 1)

并且，根据您的输入，我们得到370，等于

((((9+1)*2)+8)*3+7)*4+6

【讨论】：

非常感谢您的回答！ 1代表x的初始值，对吧？
正确。现在必须有一个初始值，因为我们的“累加器”（x 参数）将是单个值，而我们的元素（y 参数）是一个元组。由于它们看起来不再一样，我们需要一个明确的初始值。
谢谢！！这真的很有帮助！

【解决方案2】：

您可以使用 zip 作为输入并使用 1 作为初始值：

from functools import reduce

A = [1,2,3,4]
B = [9,8,7,6]

r = reduce(lambda r,ab: r*ab[0]+ab[1],zip(A,B),1)
print(r) # 370

【讨论】：