使用numpy，如何生成一个数组，其中每个索引的值是从0到第二个数组中相同索引的值的总和？

Question

给出一个numpy数组a = [3,5,7]

如何有效地生成第二个数组b，其中b [i] = numpy.Sum（a [0：i]？

我已经查看了numpy文档，但解决方案并没有向我跳出来...

预期输出为b = [3,8,15]

任何想法都会被感激！！！

谢谢

道格

Answer 1

您似乎想要numpy的cumsum函数在这里：

a=np.array([3,5,7])

In [1]: np.cumsum(a)
Out[1]: array([ 3,  8, 15], dtype=int32)