假设我有一个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
答案 0 :(得分:6)
像这样使用切片:
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])