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

时间:2015-09-07 12:52:58

标签: python numpy

假设我有一个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

1 个答案:

答案 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])