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维矩阵

1 个答案: