Adding and Subtracting Matrices in Python

numpy matrix operations | identity() function

Last Updated : 21 Feb, 2019

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 :

Python3

# 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)

Output :

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

Code #2 :

Python3

# 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)

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]]

Adding and Subtracting Matrices in Python

jana_sayantan

Improve

Article Tags :

Practice Tags :

python

numpy matrix operations | identity() function

Similar Reads

Introduction

Creating NumPy Array

NumPy Array Manipulation

Matrix in NumPy

Operations on NumPy Array

Reshaping NumPy Array

Indexing NumPy Array

Arithmetic operations on NumPyArray

Linear Algebra in NumPy Array

Thank You!

What kind of Experience do you want to share?