首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >问答首页 >如何在列表中存储reduce函数的每个迭代?

如何在列表中存储reduce函数的每个迭代?
EN

Stack Overflow用户
提问于 2020-02-13 02:35:28
回答 2查看 209关注 0票数 3

在下面的代码中,输出是38,我想要一个包含输出[34,36,38]的单独列表。

代码语言:javascript
代码运行次数:0
运行
复制
from functools import *
nums = [0, 34, 2, 2]
sum_num = reduce(lambda a, b : a+b, nums)

当reduce函数添加034时,我需要将这个值附加到一个单独的列表中,现在在第二次迭代中,我需要将34 + 2附加到列表中。最后,38将被追加到列表中。我需要添加什么代码才能获得所需的输出?

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2020-02-13 02:44:28

您需要一个不同的函数。itertools.accumulate()生成functools.reduce()在幕后生成的所有中间结果:

代码语言:javascript
代码运行次数:0
运行
复制
>>> from itertools import accumulate
>>> nums = [0, 34, 2, 2]
>>> list(accumulate(nums))
[0, 34, 36, 38]

默认使用加法。或者你可以传递任何你想要的其他双参数函数:

代码语言:javascript
代码运行次数:0
运行
复制
>>> list(accumulate(nums, lambda a, b: a + b)) # same as the default
[0, 34, 36, 38]
>>> list(accumulate(nums, lambda a, b: a + 2*b))
[0, 68, 72, 76]

如果你不想要开头的0,你必须自己去掉它;例如,

代码语言:javascript
代码运行次数:0
运行
复制
>>> f = accumulate(nums)
>>> next(f)  # throw out first result
0
>>> list(f)  # and make a list out of what remains
[34, 36, 38]
票数 6
EN

Stack Overflow用户

发布于 2020-02-13 02:45:53

根据the docs的说法,reduce函数大致相当于:

代码语言:javascript
代码运行次数:0
运行
复制
def reduce(function, iterable, initializer=None):
    it = iter(iterable)
    if initializer is None:
        value = next(it)
    else:
        value = initializer
    for element in it:
        value = function(value, element)
    return value

因此,要在整个过程中获得每个总和,我们可以使用以下函数:

代码语言:javascript
代码运行次数:0
运行
复制
def reduce(function, iterable):
    it = iter(iterable)
    value = next(it)
    values = []
    for element in it:
        value = function(value, element)
        values.append(value)
    return values

(由于未使用initializer参数,因此得到简化)

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/60194925

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档