对于每个元素，如何在 mxn 矩阵中获取最近的 r 元素

Question

我有一个名为arr 的mxn 矩阵。对于arr 中的每个元素，我想查看该元素的 r 个单位（包括对角线）内的所有元素（即子矩阵）。例如如果 r=1，则子矩阵应该是围绕当前像素的 3x3 正方形。子矩阵不一定必须是正方形（例如，如果所讨论的元素位于矩阵的边缘，则不需要）。

我尝试了以下方法，但成功有限：

r=1
for i in range(m):
    for j in range(n): 
        print(arr[max(0,i-r-1):min(i+r+1,m), max(0,j-r-1):min(j+r+1,n)])

Answer 1

r = 1
for i in range(m):
    for j in range(n):
        start_y = i-r
        end_y = i+ r + 1

        start_x = j - r
        end_x = j + r + 1

        if start_y < 0:
            start_y = 0

        elif end_y >= m:
            end_y = m

        if start_x < 0:
            start_x = 0

        elif end_x >= n:
            end_x = n

        print(arr[start_y: end_y, start_x: end_x])