numpy矩阵操作的ident()函数

numpy.matlib.ident()是另一个在numpy中进行矩阵运算的函数。它返回一个给定输入大小的正方形同位矩阵。

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

参数 :
n : [int] 输出矩阵的行和列的数量。
dtype : [可选] 希望输出的数据类型。

返回： n x n 矩阵，其主对角线设置为1，所有其他元素为0。

代码#1：

# 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 matrix :  
[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]

代码#2：

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