python - 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

しかし、numpy の sum 関数は "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の 3 つの要素の合計を表示します。

Answer 2

これでうまくいくはずです：

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

Python では、インデックスはゼロベースであり、終了インデックスは非包括的であることに注意してください。