Python numpy matrix eye()用法及代碼示例

numpy.matlib.eye()是另一個用於在numpy中進行矩陣運算的函數。它返回一個矩陣，對角線上有一個，其他位置為零。

用法： numpy.matlib.eye(n, M=None, k=0, dtype=’float’, order=’C’)

參數：
n :[int]輸出矩陣中的行數。
M :[int，可選]輸出矩陣中的列數，默認為n。
k :[int，可選]對角線的索引。 0表示主對角線，正值表示上對角線，負值表示下對角線，默認為0。
dtype :[可選]所需的輸出數據類型。
order:在內存中是以行優先(C-style)還是列主要(Fortran-style)順序存儲多維數據。

Return :一個n x M矩陣，其中所有元素均等於0，除了k-th對角線(其值等於1)之外。

代碼1：

# Python program explaining 
# numpy.matlib.eye() function 
  
# importing matrix library from numpy 
import numpy as geek 
import numpy.matlib 
  
# desired 3 x 3 output matrix  
out_mat = geek.matlib.eye(3, k = 0)  
print ("Output matrix:", out_mat)

輸出：

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

代碼2：

# Python program explaining 
# numpy.matlib.eye() function 
  
# importing numpy and matrix library 
import numpy as geek 
import numpy.matlib 
  
# desired 4 x 5 output matrix  
out_mat = geek.matlib.eye(n = 4, M = 5, k = 1, dtype = int)  
print ("Output matrix:", out_mat)

輸出：

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