標識矩陣

Question

當我在學習python時，我遇到了一個問題，一直無法確定哪裏出了問題。請看下面的python代碼。

def is_identity_matrix(matrix): 
    n = len(matrix) 
    if n != len(matrix[0]): 
     return False 
    i = 0 
    while i < n: 
     j = 0 
     while j < n: 
      if matrix[i][j] == matrix[i][i]: 
       if matrix[i][j] != 1: 
        return False 
      else: 
       if matrix[i][j] != 0: 
        return False 
      j+=1 
     i+=1 
    return True 

matrix6 = [[1,0,0,0], 
      [0,1,0,1], 
      [0,0,1,0], 
      [0,0,0,1]] 

print is_identity_matrix(matrix6) 

此輸出True代替False因爲這個問題是要求我們檢查單位矩陣只有在中間的對角線應有1，剩下的應該是0。由於matrix[1][3]不等於1 ，輸出應該是錯誤的。任何人都可以幫忙嗎？非常感謝！

Answer 1

你應該比較i == j而不是matrix[i][j] == matrix[i][i]。

這裏的版本使用for而不是while。

def is_identity_matrix(matrix): 
    n = len(matrix) 
    if n != len(matrix[0]): 
     return False 
    for i in range(n): 
     for j in range(n): 
      if i == j: 
       if matrix[i][j] != 1: 
        return False 
      else: 
       if matrix[i][j] != 0: 
        return False 
    return True 

甚至（使用product從itertools），這節省了縮進層次，看起來更可讀的對我說：現在

from itertools import product 

def is_identity_matrix(matrix): 
    n = len(matrix) 
    if n != len(matrix[0]): 
     return False 
    for i, j in product(range(n), range(n)): 
     if i == j: 
      if matrix[i][j] != 1: 
       return False 
     else: 
      if matrix[i][j] != 0: 
       return False 
    return True 

Answer 2

就像弘主角說，問題是在該行：

if matrix[i][j] == matrix[i][i]: 

如果一個單元格的值在對角線上，在這裏我假設你要檢查，如果你是對的值相匹配您正在檢查或離開對角線。

該行應爲：

if i == j: 

Answer 3

，如果你還沒有學會範圍（），就像我一樣， 這裏是我的解決方案：

def is_identity_matrix(matrix): 
    #Write your code here 
    n=len(matrix) 
    if n==len(matrix[0]): 
     i=0 
     while i<n: 
      j=0 
      while j<n: 
       if j==i: 
        if matrix[i][j]!=1: 
         return False 
       else: 
        if matrix[i][j]!=0 or matrix[j][i]!=0: 
         return False 
       j=j+1 
      i=i+1 
     else: 
      return True 
    else: 
     return False