溫馨提示×

如何在python中使用eye函數(shù)

小樊
101
2024-09-08 21:50:08
欄目: 編程語言

eye 函數(shù)是 NumPy 庫中的一個(gè)函數(shù)，用于創(chuàng)建一個(gè)對角線上為1，其余元素為0的二維數(shù)組（也稱為單位矩陣）

首先，確保已經(jīng)安裝了 NumPy。如果沒有安裝，可以使用 pip 進(jìn)行安裝：

pip install numpy

然后，在 Python 代碼中使用 eye 函數(shù)：

import numpy as np

# 創(chuàng)建一個(gè) 3x3 的單位矩陣
identity_matrix = np.eye(3)

print(identity_matrix)

輸出結(jié)果將會是：

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

eye 函數(shù)還有一些可選參數(shù)，例如設(shè)置對角線上的值、指定矩陣的形狀等。更多關(guān)于 eye 函數(shù)的信息和用法，請參考 NumPy 官方文檔：https://numpy.org/doc/stable/reference/generated/numpy.eye.html

0 踩