累积列表元素，同时为每个累积添加常数

Question

我在一个列表中积累值，使一个值是所有先前值的总和，使用np.cumsum():

l=[4,2,1,3]
c = np.cumsum(l)
print(c)

[4 6 7 10]

但是我还想将1添加到每个计算中，因此结果看起来像:

[4 7 9 13]

最简单的方法是什么?

Answer 1

给你:

res = c + np.arange(len(c))