Python numpy matrix identity()用法及代码示例

numpy.matlib.identity()是另一个用于在numpy中进行矩阵运算的函数。它返回给定输入大小的平方单位矩阵。

用法： numpy.matlib.identity(n, dtype=None)

参数：
n :[int]输出矩阵中的行数和列数。
dtype :[可选]所需的输出数据类型。

Return :n x n矩阵，其主对角线设置为1，所有其他元素设置为零。

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

相关用法

注：本文由纯净天空筛选整理自jana_sayantan大神的英文原创作品 numpy matrix operations | identity() function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。