Python 2D 数组在行和列上搜索相邻单元格元素

Question

我有一个依赖于用户输入的二维数组，每个元素只有 0 和 1。示例： matric=[[1,0,1],[1,0,0],[1,0,1] 代表病毒感染，如果用户在第[1,0,1]行输入就会感染[1 ,0,1]=>[1,1,1] 因为离 1 最近的单元格会感染。我想从用户输入中获取所有行和列，如果该值是 0，即值 1 旁边或之前的值。 这是我的一些感染细胞代码：

def isInfected(grid, r, c):
    return grid[r][c] == 1

功能检查获得下一次感染：

def getNextInfectedCells(grid):
    row = len(grid)
    column = len(grid[0])
    result = []

    for r in range(row):
        result.append([])
        for c in range(column:
            if(r==0 or r==row-1) or (c==0 or c==column-1):
                result.append(grid[r][c])
            elif grid[r][c]==0:
                result[r].append(0)
            else:
                result[r].append(grid[r][c])
        for line in result:
            print(line)

怎么了！它不起作用 最好的尊重！

Answer 1

我不确定我是否理解问题，但这段代码应该可以工作。

user_input = [[1,0,1],[1,0,0],[1,0,1]]

for row in user_input:
    if 1 in row:
        infected = row.index(1) + 1
        row[infected] = 1