使用累加函数的python中的前缀求和数组

给定一个数组，我们必须使用accumulate function.itertools.accumulate（iterable [，func]）模块函数来构造所有的前缀并返回迭代器。因此，只能通过截断流的函数或循环来访问它们。进行迭代，以返回累加的总和。元素可以是任何可加类型，包括小数或小数。如果提供了可选的函数参数，则它应该是两个参数的函数，并且将使用它代替加法。

示例

Input
Data = [1, 0, 2, 3, 5]
>>> list(accumulate(data)) # running summation
Output
[1, 1, 3, 6, 11]

算法

Step 1: Create list.
Step 2: use list(accumulate( ))) function, its return running total.
Step 3: display total.

范例程式码

# Python program to print prefix
# sum array using accumulate function from itertools import accumulate
def summation(A):
   print ("The List after Summation ::>", list(accumulate(A)))
      # Driver program
      if __name__ == "__main__":
      A=list()
      n=int(input("Enter the size of the First List ::"))
      print("Enter the Element of First List ::")
      for i in range(int(n)):
      k=int(input(""))
      A.append(k)
summation(A)

输出结果

Enter the size of the First List ::5
Enter the Element of First List ::
1
2
3
4
5
The List after Summation ::> [1, 3, 6, 10, 15]