Python中使用累积函数为和数组添加前缀

我们得到一个数组，找到给定数组的前缀和。

例子：

Input : arr = [1, 2, 3]
Output : sum = [1, 3, 6]

Input : arr = [4, 6, 12]
Output : sum = [4, 10, 22]

前缀和是给定序列的部分和的序列。例如，序列{a, b, c, ...}的累积和为a、a+b、a+b+c等。我们可以在Python中使用累积(iterable)方法快速解决这个问题。

# function to find cumulative sum of array
from itertools import accumulate
  
def cumulativeSum(input):
      print ("Sum :", list(accumulate(input)))
  
# Driver program
if __name__ == "__main__":
    input = [4, 6, 12]
    cumulativeSum(input)

输出：

Sum = [4, 10, 22]