Question

How to generate an identity matrix in Python

Answer 1

import numpy as np

print(np.identity(5))



'''
run:

[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]

'''

 

70+ SQL courses for beginners and professionals

Answer 2

size = 5 
identity_matrix = [ [1 if num == index else 0 for index in range(size)]
                  for num in range(size) ]
          
for row in identity_matrix:
    print(row)   
    

                  
                  

'''
run:

[1, 0, 0, 0, 0]
[0, 1, 0, 0, 0]
[0, 0, 1, 0, 0]
[0, 0, 0, 1, 0]
[0, 0, 0, 0, 1]

'''

 

70+ SQL courses for beginners and professionals