来自 m×n 矩阵的 5×5 维矩阵

Question

我有一个 m×n 维矩阵，我需要通过索引i、j 获得 5×5 维矩阵。结果应该是这样的：

res = [[arr[i-2][j-2], arr[i-2][j-1]], arr[i-2][j  ], arr[i-2][j+1], arr[i-2][j+2]]
       [arr[i-1][j-2], arr[i-1][j-1]], arr[i-1][j  ], arr[i-1][j+1], arr[i-1][j+2]],
       [arr[i  ][j-2], arr[i  ][j-1]], arr[i  ][j  ], arr[i  ][j+1], arr[i  ][j+2]],
       [arr[i+1][j-2], arr[i+1][j-1]], arr[i+1][j  ], arr[i+1][j+1], arr[i+1][j+2]],
       [arr[i+2][j-2], arr[i+2][j-1]], arr[i+2][j  ], arr[i+2][j+1], arr[i+2][j+2]]]

问题是数组的索引。 i、j 是随机数，所以可以是0, 0 或m-1, n-1。在这种情况下，[arr[i-2][j-2] 或 [arr[i+2][j+2] 等有问题的值必须替换为 'x'。有什么干的方法可以得到结果吗？

Answer 1

我想这样的事情会起作用：

def give_matrix(arr,i,j):
    res = []
    for ii in range(i-2,i+3): 
        inner_res = []
        for jj in range(j-2,j+3):
            if (ii-2<0 or ii+3>n or jj-2<0 or jj+3>n): # manage borders
                inner_res.append('x')
            else:
                inner_res.append(arr[ii][jj])
        res.append(inner_res)
    return res

Livecode