numpy.eye
是 NumPy 庫中的一個函數,用于創建一個對角線上元素為 1,其余元素為 0 的二維數組(也稱為單位矩陣)
numpy.eye(N, M=None, k=0, dtype=float, order='C', *, like=None)
參數說明:
N
:輸出矩陣的行數。必需參數。M
:輸出矩陣的列數。如果未指定,默認值為 N
,即創建一個方陣。k
:指定對角線的位置。正值表示上對角線,負值表示下對角線,0 表示主對角線。默認值為 0。dtype
:輸出矩陣的數據類型。默認值為 float
。order
:輸出矩陣的內存布局。可以是 'C'
(按行優先存儲)或 'F'
(按列優先存儲)。默認值為 'C'
。like
:用于指定輸出矩陣的類型和屬性。如果提供了此參數,將根據 like
的類型和屬性創建輸出矩陣。示例:
import numpy as np
# 創建一個 3x3 的單位矩陣
identity_matrix = np.eye(3)
print(identity_matrix)
# 創建一個 4x4 的單位矩陣,其中對角線上的元素為 2
scaled_identity_matrix = np.eye(4, dtype=int) * 2
print(scaled_identity_matrix)
# 創建一個 4x4 的矩陣,其中第 2 條對角線上的元素為 1
diagonal_matrix = np.eye(4, k=1)
print(diagonal_matrix)
輸出:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
[[2 0 0 0]
[0 2 0 0]
[0 0 2 0]
[0 0 0 2]]
[[0. 1. 0. 0.]
[0. 0. 1. 0.]
[0. 0. 0. 1.]
[0. 0. 0. 0.]]