python - 如何就地获取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])