python - 处理 numpy 数组中选定范围的值

Question

假设我有一个 5x5 数组:

import numpy as np
arr = np.random.rand(5,5)

如果我想对整个数组求和，我可以简单地拥有:

np.sum(arr)

我将如何对由左上角 (2,2) 和右下角 (4,3) 定义的框中的值求和？

如果这不是很清楚，我想对下面数组中的粗体 x 求和:

X X X X X

X X X X X

X X X X X

X X X X X

X X X X X

Answer 1

像这样使用切片:

import numpy as np
arr = np.random.rand(5,5)

# Top left 2*2 grid
np.sum(arr[:2, :2])

要对图表中的数组求和，请使用:

np.sum(arr[1:4, 1:3])