使用 accumulate 函数在 Python 中对数组进行前缀求和
pythonprogrammingserver side programming
给定一个数组,我们必须使用 accumulate 函数对数组进行前缀求和。itertools.accumulate(iterable[, func]) 模块函数都构造并返回迭代器。因此,它们只能由截断流的函数或循环访问。创建一个返回累积总和的迭代器。元素可以是任何可添加的类型,包括 Decimal 或 Fraction。如果提供了可选函数参数,它应该是两个参数的函数,并且将使用它来代替加法。
示例
输入 数据 = [1, 0, 2, 3, 5] >>> list(accumulate(data)) # 运行求和 输出 [1, 1, 3, 6, 11]
算法
步骤 1:创建列表。 步骤 2:使用 list(accumulate( ))) 函数,其返回运行总数。 步骤 3:显示总数。
示例代码
# Python 程序打印前缀 # 使用 accumulate 函数对数组求和 from itertools import accumulate def summation(A): print ("求和后的列表 ::>", list(accumulate(A))) # 驱动程序 if __name__ == "__main__": A=list() n=int(input("输入第一个列表的大小 ::")) print("输入第一个列表的元素 ::") for i in range(int(n)): k=int(input("")) A.append(k) summation(A)
输出
输入第一个列表的大小 ::5 输入第一个列表的元素 :: 1 2 3 4 5 求和后的列表 ::> [1, 3, 6, 10, 15]