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: 配列は実際には 2 次元配列であるため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])