使用累加功能在python中添加前缀sum数组
给定一个数组,我们必须使用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]