numpy matrix operations | identity() function

numpy.matlib.identity() is another function for doing matrix operations in numpy. It returns a square identity matrix of given input size.

Syntax : numpy.matlib.identity(n, dtype=None)

Parameters :
n : [int] Number of rows and columns in the output matrix.
dtype : [optional] Desired output data-type.



Return : n x n matrix with its main diagonal set to one, and all other elements zero.

Code #1 :

filter_none

edit
close

play_arrow

link
brightness_4
code

# Python program explaining
# numpy.matlib.identity() function
  
# importing matrix library from numpy
import numpy as geek
import numpy.matlib
  
# desired 3 x 3 output square identity matrix 
out_mat = geek.matlib.identity(3) 
print ("Output matrix : ", out_mat) 

chevron_right






                        filter_none









Output : 


Output matrix :  
[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]





 


Code #2 : 











                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Python program explaining 


# numpy.matlib.identity() function 


  


# importing numpy and matrix library 


import numpy as geek 


import numpy.matlib 


  


# desired 5 x 5 output square identity matrix  


out_mat = geek.matlib.identity(n = 5, dtype = int)  


print ("Output matrix : ", out_mat)  



















                        chevron_right







                        filter_none









Output : 


Output matrix :  
[[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]]






          
          
          
          

            

    

        My Personal Notes
        arrow_drop_up
    

    

        
        

            
            
        

    


Recommended Posts:
jana_sayantan
Check out this Author's contributed articles.
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on  the "Improve Article" button below.