时间戳

首页 > 解决方案 > 我可以在reduce函数中添加一个迭代器吗？

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

问题描述

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

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

但是，如果我有另一个 list 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函数中。我怎么能这样做？非常感谢！

标签： pythoniteratorreduce

解决方案

这看起来像是一份工作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

推荐阅读