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]從矩陣，我怎麼能指望周圍的spesific點字符？例如，假設我選擇W [4] [1]，即b。我可以看到那邊有三個a。但是，我怎樣才能通過編碼來確定它呢？我最終弄亂了代碼的和平，並且意識到如果我們改變矩陣的維度，它就無法工作。問題在於，如果我選擇接近輪的點，W[y-1][(x):(x+1)].count("a") - 思維的種類不起作用。

再次，幫助將不勝感激！謝謝！

UPDATE：

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 

取出if (a,b)!=(0,0)如果要算在廣場本身的字符。即，用方形0,0你是否在那裏計算'a'？

Answer 2

的說明，請參見代碼註釋：

from itertools import product 

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_chars(W, x, y, ch): 
    return sum(W[x+dx][y+dy] == ch 
        for dx, dy in product([-1,0,1], [-1,0,1]) 
         if 0 <= x+dx < len(W) and 0 <= y+dy < len(W[0])) 

Answer 3

關於以問題與原矩陣：

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, x, y): 
    occ = 0 
    for n in range(max(0, y - 1), min(len(W), y + 2)): 
     for m in range(max(0, x - 1), min(len(W[0]), x + 2)): 
      if W[n][m] == "a": 
       occ += 1 
    return occ 

print count_matrix(W, 0, 0) 

retunrs 3。唯一剩下的問題是如何排除廣場本身的價值？

編輯：

def count_matrix(W, x, y): 
occ = 0 
for n in range(max(0, y - 1), min(len(W), y + 2)): 
    for m in range(max(0, x - 1), min(len(W[0]), x + 2)): 
     if n == y and m == x: 
      pass 
     else: 
      if W[n][m] == "a": 
       occ += 1 
return occ 

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']] 

print count_matrix(W, 1, 4) 

返回3預期。

print count_matrix(W, 0, 0) 

如預期的那樣返回2。這是好的做法嗎？