python - 检查矩阵中 "b"周围的值

Question

还是扫雷。 我找到了一种方法来做到这一点，但我知道必须有一种简化的方法来做到这一点。我必须在矩阵中放置一个数字来表示它周围有多少炸弹(“b”)。这就是我所拥有的，我知道必须有更短的方法。

def check(y,x):
    if ((y < 0) or (y >= len(mat1)) or (x < 0) or (x >= len(mat1))):
        return (False)
    else:
        return mat1[y][x]

def addscores():
    for x in range(len(mat1)):
        for y in range(len(mat1)):
            if mat1[y][x] != "b":
                if check(y-1,x-1) == "b" or check(y,x-1) == "b" or check(y+1,x-1) == "b" or check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) =="b":
                    mat1[y][x] = 1
                if check(y-1,x-1) == "b":
                    if check(y,x-1) == "b" or check(y+1,x-1) == "b" or check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y,x-1) == "b":
                    if check(y+1,x-1) == "b" or check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y+1,x-1) == "b":
                    if check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                         mat1[y][x] = 2
                if check(y+1,x) == "b":
                    if check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y+1,x+1) == "b":
                    if check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y,x+1) == "b":
                    if check(y-1,x+1) == "b" or check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y-1,x+1) == "b":
                    if check(y-1,x) == "b":
                        mat1[y][x] = 2
                if check(y-1,x-1) == "b":
                    if check(y,x-1) == "b":
                        if check(y+1,x-1) == "b" or check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                            mat1[y][x] = 3
                    if check(y+1,x-1) == "b":
                        if check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                            mat1[y][x] = 3
                    if check(y+1,x) == "b":
                        if check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                            mat1[y][x] = 3
                    if check(y+1,x+1) == "b":
                        if check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                            mat1[y][x] = 3
                    if check(y,x+1) == "b":
                        if check(y-1,x+1) == "b" or check(y-1,x) == "b":
                            mat1[y][x] = 3
                if check(y-1,x-1) == "b":
                    if check(y,x-1) == "b":
                        if check(y+1,x-1) == "b":
                            if check(y+1,x) == "b" or check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                                mat1[y][x] = 4
                    if check(y+1,x-1) == "b":
                        if check(y+1,x) == "b":
                            if check(y+1,x+1) == "b" or check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                                mat1[y][x] = 4
                    if check(y+1,x) == "b":
                        if check(y+1,x+1) == "b":
                            if check(y,x+1) == "b" or check(y-1,x+1) == "b" or check(y-1,x) == "b":
                                mat1[y][x] = 4
                    #ETC

Answer 1

def check(y,x):
    if ((y < 0) or (x < 0) or (y >= len(mat1)) or (x >= len(mat1))):
        return False
    else:
        if mat1[y][x] == 'b':
            return 1
        else:
            return 0

def check_all(y,x):
    if mat1[y][x] != 'b':        
        return sum([check(y + yy, x + xx) for xx in range(-1,2) for yy in range(-1,2)])
    else:
        return 'b'

def addscores():
    for x in range(len(mat1)):
        for y in range(len(mat1)):
            mat1[y][x] = check_all(y,x)

如果我正确理解了您所做的事情，那么这段代码(尤其是 check_all 函数)应该可以解决您的问题。你是对的，肯定有一种更短的方法来做到这一点，使用循环(或在本例中是列表推导式)，而不是必须单独写出每个检查。

我尽可能保留了您的代码，因为我没有足够的上下文来知道进行更改是否会破坏任何内容。