numpy.identity() in Python
Last Updated :
05 Aug, 2021
Improve
numpy.identity(n, dtype = None) : Return a identity matrix i.e. a square matrix with ones on the main diagonal.
Parameters : n : [int] Dimension n x n of output array dtype : [optional, float(by Default)] Data type of returned array.
Returns : identity array of dimension n x n, with its main diagonal set to one, and all other elements 0.
Example:
Python
# Python Programming illustrating # numpy.identity method import numpy as geek # 2x2 matrix with 1's on main diagonal b = geek.identity( 2 , dtype = float ) print ( "Matrix b : \n" , b) a = geek.identity( 4 ) print ( "\nMatrix a : \n" , a) |
Output :
Matrix b : [[ 1. 0.] [ 0. 1.]] Matrix a : [[ 1. 0. 0. 0.] [ 0. 1. 0. 0.] [ 0. 0. 1. 0.] [ 0. 0. 0. 1.]]
Note :
These codes won’t run on online-ID. Please run them on your systems to explore the working.