eye
函數是 NumPy 庫中用于創建單位矩陣(identity matrix)的一個函數
以下是 eye
函數與其他矩陣函數的比較:
numpy.eye(N, M=None, k=0, dtype=float)
:創建一個 NxM 的單位矩陣,其中 N 是行數,M 是列數(默認為 N),k 是對角線索引(0 表示主對角線,正數表示上對角線,負數表示下對角線)。dtype 參數指定矩陣元素的數據類型。import numpy as np
I = np.eye(3)
print(I)
輸出:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
numpy.zeros((N, M), dtype=float)
:創建一個 NxM 的全零矩陣。Z = np.zeros((3, 3))
print(Z)
輸出:
[[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]]
numpy.ones((N, M), dtype=float)
:創建一個 NxM 的全一矩陣。O = np.ones((3, 3))
print(O)
輸出:
[[1. 1. 1.]
[1. 1. 1.]
[1. 1. 1.]]
numpy.diag(v, k=0)
:從給定向量 v 創建一個對角矩陣,k 參數表示對角線索引。D = np.diag([1, 2, 3])
print(D)
輸出:
[[1 0 0]
[0 2 0]
[0 0 3]]
總結:eye
函數專門用于創建單位矩陣,而其他矩陣函數可以創建全零、全一或自定義對角線的矩陣。在需要單位矩陣時,使用 eye
函數更直觀和高效。