求和返回Python中的向量

如何解决求和返回Python中的向量

我有一个数组，例如：[10,6,4,12]。 我有兴趣找到一个累积和的向量，即： [10、16、20、32]。 显而易见的方法是使用for循环：

r = []
for ind in range(4):
  r.append(s[ind])  # s is [10,12]
r = cumsum(r)

但是，这似乎效率很低。我想问一下是否有一个预定义的函数，或者我应该在cumsum中指定特定的参数。

解决方法

您可以使用itertools.accumulate

的多种方式，例如np.cumsum或python 3.2+

通过Itertool：

l = [10,6,4,12]

from itertools import accumulate

print(list(accumulate(l)))

输出：

[10,16,20,32]

使用numpy：

import numpy as np

print(np.cumsum(l))

输出：

[10,32]

,

Libless解决方案：

s = [10,12]
r = [sum(s[:i+1]) for i in range(len(s))]

输出：

[10,32]