python：numpy數組的多種算術運算

Question

我有此代碼（需要從當前值中減去先前元素的總和）：

arr = np.zeros((N,M,T))
for it in xrange(T):
     sum_arr = np.zeros((M,N))
     for tt in xrange(it): sum_arr += arr[:,:,tt]
     arr[:,:,it] -= sum_arr

問題：是否可以用pythonic方式編寫此代碼（首選一行）？ 提前謝謝。

Answer 1

我認為您至少可以更有效地完成總和：

arr = np.zeros((N, M, T))
for it in xrange(T):
    arr[:,:,it] -= np.sum(arr[:,:,:it], axis=2)

這幾乎是1班輪：

for it in xrange(T): arr[:,:,it] -= np.sum(arr[:,:,:it], axis=2)

我假設您的實際數據arr並非全為零-否則，總和將為零數組，然后從零數組中減去零，剩下...和零數組（這不是很有趣） 。