使用 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]

相关文章