计算二维列表 python 周围正方形中有多少个符号

Question

我在 python 课程的介绍中有一个作业，我必须制作一个 function 使用循环遍历二维列表，当给定 x 和 y 坐标时，它会计算所选方格周围有多少“N”。 x 和 y 可以在任何地方，并且不允许它们越过列表的边界。 由于某种原因，课程材料似乎非常缺乏，我找不到任何可以帮助我入门的东西。 我应该如何 go 关于使 function 工作？

function 的参数必须是（x，y，列表） function 应该假设您不在“N”上，但如果您是，它也算一个。

如果有人可以给我一些关于如何开始的提示，那就太好了。

list = [['N', ' ', ' ', ' ', ' '],
        ['N', 'N', 'N', 'N', ' '],
        ['N', ' ', 'N', ' ', ' '],
        ['N', 'N', 'N', ' ', ' '],
        [' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ']]

     

Answer 1

从此代码中，您可以计算列表中有多少 N：

list_ = [['N', ' ', ' ', ' ', ' '],
        ['N', 'N', 'N', 'N', ' '],
        ['N', ' ', 'N', ' ', ' '],
        ['N', 'N', 'N', ' ', ' '],
        [' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ']]
count = 0
for a in list_:
    for b in a:
        if b == 'N':
            count += 1
print(count)

Answer 2

从问题中不清楚提供的 x 和 y 坐标是否是边缘周围的元素，相邻元素少于 8 个。

如果 x 坐标的范围为1到len(row) - 1并且 y 坐标的范围为1到len(column)-1那么您可以循环遍历 2d 列表中的每一行，然后遍历其中的每个元素行，然后检查该元素是否等于'N' 。 这可以这样解决：

def adjacent_count(x, y, list_2d):
    count = 0
    for row in list_2d[x-1:x+2]:
        for element in row[y-1:y+2]:
            if element == 'N':
                count += 1
    return count

或者在一行中使用列表理解：

def adjacent_count(x, y, list_2d):
    return sum([1 if element == 'N' else 0 for row in list_2d[x-1:x+2] for element in row[y-1:y+2]])

编辑：现在知道 x 和 y 坐标的范围可以从0到len(rows)和len(cols)并且 function 不应该换行到数组的另一侧，可以使用以下代码。 代码现在在尝试访问之前检查下一个数组元素是否超出列表的范围。

def adjacent_count(x, y, list_2d):
    no_rows, no_cols = len(list_2d), len(list_2d[0])
    count = 0
    for i in range(x-1, x+2):
        for j in range(y-1, y+2):
            if i < no_cols and j < no_rows:
                if list_2d[j][i] == 'N':
                    count += 1
    return count