eye
函數是 NumPy 庫中的一個函數,用于創建一個對角線上為1,其余元素為0的二維數組(也稱為單位矩陣)
首先,確保已經安裝了 NumPy。如果沒有安裝,可以使用 pip 進行安裝:
pip install numpy
然后,在 Python 代碼中使用 eye
函數:
import numpy as np
# 創建一個 3x3 的單位矩陣
identity_matrix = np.eye(3)
print(identity_matrix)
輸出結果將會是:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
eye
函數還有一些可選參數,例如設置對角線上的值、指定矩陣的形狀等。更多關于 eye
函數的信息和用法,請參考 NumPy 官方文檔:https://numpy.org/doc/stable/reference/generated/numpy.eye.html