来自 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

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

5×5 dimensional matrix from m×n matrix

python

algorithm

list

dry

sublist