使用np.cumsum累加列表元素时如何给每步累加值添加固定常数

实现方法 #

直接在np.cumsum()的计算结果上，叠加一个和输入序列等长的从0开始的等差序列即可，第i个位置（从0计数）累计需要额外加i次常数1，刚好和等差序列的取值完全匹配。
示例代码如下：

import numpy as np
l = [4, 2, 1, 3]
c = np.cumsum(l) + np.arange(len(l))
print(c)

输出结果：

[ 4 7 9 13]

内容的提问来源于stack exchange，提问作者Encomium