如何获得numpy数组的累积和

Question

我想计算积分图像。例如

a=array([(1,2,3),(4,5,6)])
b = a.cumsum(axis=0)

这将生成另一个数组b。我可以就地执行cumsum。如果不 。有没有其他方法可以做到这一点

Answer 1

您必须传递参数out：

np.cumsum(a, axis=1, out=a)

OBS：您的数组实际上是一个二维数组，因此您可以使用axis=0对行进行求和，并axis=1沿着列求和。

Answer 2

直接使用numpy numpy.cumsum(a)：

尝试此操作

a=array([(1,2,3)])
b = np.cumsum(a)
print b
>>array([1,3,6])