itertools.accumulate() versus functools.reduce()

2019-03-24 19:13发布

In Python 3.3, itertools.accumulate(), which normally repeatedly applies an addition operation to the supplied iterable, can now take a function argument as a parameter; this means it now overlaps with functools.reduce(). With a cursory look, the main differences between the two now would seem to be:

  1. accumulate() defaults to summing but doesn't let you supply an extra initial condition explicitly while reduce() doesn't default to any method but does let you supply an initial condition for use with 1/0-element sequences, and
  2. accumulate() takes the iterable first while reduce() takes the function first.

Are there any other differences between the two? Or is this just a matter of behavior of two functions with initially distinct uses beginning to converge over time?

3条回答
何必那么认真
2楼-- · 2019-03-24 19:36

itertools.accumulate is like reduce but returns a generator* instead of a value. This generator can give you all the intermediate step values. So basically reduce gives you the last element of what accumulate will give you.

*A generator is like an iterator but can be iterated over only once.

查看更多
闹够了就滚
3楼-- · 2019-03-24 19:41

It seems that accumulate keeps the previous results, whereas reduce (which is known as fold in other languages) does not necessarily.

e.g. list(accumulate([1,2,3], operator.plus)) would return [1,3,6] whereas a plain fold would return 6

Also (just for fun, don't do this) you can define accumulate in terms of reduce

def accumulate(xs, f):
    return reduce(lambda a, x: a + [f(a[-1], x)], xs[1:], [xs[0]]) 
查看更多
干净又极端
4楼-- · 2019-03-24 19:53

You can see in the documentation what the difference is. reduce returns a single result, the sum, product, etc., of the sequence. accumulate returns an iterator over all the intermediate results. Basically, accumulate returns an iterator over the results of each step of the reduce operation.

查看更多
登录 后发表回答