30秒学会 Python 片段 · 2022年8月11日

30秒学会 Python 片段 – Partial sum list

Creates a list of partial sums.

  • Use itertools.accumulate() to create the accumulated sum for each element.
  • Use list() to convert the result into a list.

代码实现

from itertools import accumulate

def cumsum(lst):
  return list(accumulate(lst))

使用样例

cumsum(range(0, 15, 3)) # [0, 3, 9, 18, 30]

翻译自:https://www.30secondsofcode.org/python/s/cumsum