Question

How to check if a matrix is an identity matrix (1 on the main diagonal and 0 elsewhere) in Python

Answer 1

m = [     
        [1, 0, 0],  
        [0, 1, 0],  
        [0, 0, 1]  
    ];  
      
flag = True;  
   
rows = len(m);  
cols = len(m[0]);  
   
if (rows != cols):  
    print("Not a square matrix");  
else:  
    for i in range(0, rows):  
        for j in range(0, cols):  
            if (i == j and m[i][j] != 1):  
                flag = False;  
                break;  
            if (i != j and m[i][j] != 0):  
                flag = False;  
                break;  
   
    if (flag):  
        print("Identity matrix");  
    else:  
        print("Not identity matrix");  
        
        
   
        
'''
run:

Identity matrix

'''

 

70+ SQL courses for beginners and professionals