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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。