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