eye
函數在 Python 中主要用于創建一個對角線上元素為 1,其余元素為 0 的二維數組
eye
函數可以方便地創建單位矩陣。import numpy as np
identity_matrix = np.eye(3)
print(identity_matrix)
輸出:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
eye
函數還可以用于創建具有指定對角線元素的矩陣。通過設置 k
參數,可以控制對角線的位置。import numpy as np
diagonal_matrix = np.eye(4, k=1)
print(diagonal_matrix)
輸出:
[[0. 1. 0. 0.]
[0. 0. 1. 0.]
[0. 0. 0. 1.]
[0. 0. 0. 0.]]
eye
函數創建一個對角線上元素為 1,其余元素為 0 的矩陣,然后將其與其轉置相加,得到一個對稱矩陣。import numpy as np
symmetric_matrix = np.eye(3) + np.eye(3).T
print(symmetric_matrix)
輸出:
[[2. 1. 1.]
[1. 2. 1.]
[1. 1. 2.]]
總之,eye
函數在 Python 中的適用場景非常廣泛,可以用于創建單位矩陣、對角矩陣、對稱矩陣等。