如何在python/numpy中對數組中的元素進行求和？

Question

如何求和python/numpy數組中元素的間隔？

例如，在MATLAB有可能是這樣的：

y= ones(10,20); 
y(1:5,:) = 2; 
for k = 1: 10 
    psum(k) = sum(y((1:3),k)); 
end 

但SUM函數從numpy的並不能支持 「1：3」

謝謝！

Answer 1

大概是這樣的：

y = numpy.ones((10, 20)) 
y[0:5] = 2 
psum = y[0:3].sum(axis=0) 

一個簡單的例子：

y = numpy.array([2, 6, 7, 3, 4]) 
print y[1:4].sum() 

打印16，中間的三個要素的總和。

Answer 2

這應該做的伎倆：

>>> y = np.ones((10, 20)) 
>>> y[0:5, :] = 2 
>>> psum = y[0:3, 0:10].sum(axis=0) 

注意的是Python，指數是零基礎，最終指數不包括在內。