numpy.identity() in Python

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.