Python 和矩阵

Question

我有以下矩阵:

W = [['a', 'b', 'b', 'b', 'a'],
     ['a', 'a', 'a', 'a', 'b'],
     ['a', 'b', 'a', 'b', 'b'],
     ['a', 'a', 'a', 'b', 'b'],
     ['b', 'b', 'b', 'b', 'b'],
     ['b', 'b', 'b', 'b', 'b']]

如果我从矩阵中选择 W[x][y]，如何计算该特定点周围的 a 字符？例如，假设我选择 W[4][1]，即 b。我可以看到该点周围有三个 a。但如何通过编码来确定呢？我最终得到了非常困惑的代码，并意识到如果我们改变矩阵的维度，它就不会起作用。问题还在于，如果我选择靠近回合的点，W[y-1][(x):(x+1)].count("a") -这种思维不会工作。

再次感谢您的帮助!谢谢!

更新:

W     = [['K', ' ', ' ', ' ', ' '],
         ['K', 'K', 'K', 'K', ' '],
         ['K', ' ', 'K', ' ', ' '],
         ['K', 'K', 'K', ' ', ' '],
         [' ', ' ', ' ', ' ', ' '],
         [' ', ' ', ' ', ' ', ' ']]

def count_matrix(W, y, x, ch="K"):
    ref = (-1, 0, 1)
    occ = []
    a = "K"
    b = " "
    for dy, dx in [(a, b) for a in ref for b in ref if (a,b)!=(0, 0)]:
        if (x+dx) >= 0 and (y+dy) >= 0:
            try:
                occ.append(W[x+dx][y+dy])
            except IndexError:
                pass
    return occ.count(ch)
print count_matrix(W,0,0)

这将返回0。

Answer 1

这有效:

W = [['a', 'b', 'b', 'b', 'a'],
     ['a', 'a', 'a', 'a', 'b'],
     ['a', 'b', 'a', 'b', 'b'],
     ['a', 'a', 'a', 'b', 'b'],
     ['b', 'b', 'b', 'b', 'b'],
     ['b', 'b', 'b', 'b', 'b']]

def count_matrix(W, r,c, ch):
    ref=(-1,0,1)
    matches=[]
    for dr, dc in [(a, b) for a in ref for b in ref if (a,b)!=(0,0)]:
        if r+dr>=0 and c+dc>=0:
            try:
                matches.append(W[r+dr][c+dc])
            except IndexError:
                pass
    return matches.count(ch)

print count_matrix(W,0,0,'a')   # correctly handles upper LH 
# 2
print count_matrix(W,4,1,'a') 
# 3

如果您想计算方 block 本身的字符数，请删除if (a,b)!=(0,0)。即，对于正方形0,0，你是否数过那里的“a”？