eye函數在python中的實際應用

python

小樊

105

2024-09-08 21:52:50

欄目: 編程語言

eye 函數在 Python 中主要用于創建一個對角線上元素為 1，其余元素為 0 的二維數組

以下是 eye 函數在 Python 中的一些實際應用：

import numpy as np

identity_matrix = np.eye(3)
print(identity_matrix)

輸出結果：

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

import numpy as np

custom_diagonal_matrix = np.eye(3, k=1)
print(custom_diagonal_matrix)

輸出結果：

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

import numpy as np

custom_dtype_matrix = np.eye(3, dtype=int)
print(custom_dtype_matrix)

輸出結果：

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

import numpy as np

A = np.array([[1, 2], [3, 4]])
I = np.eye(2)
B = A + I
print(B)

輸出結果：

[[2. 2.]
 [3. 5.]]

總之，eye 函數在 Python 中可以用于創建和操作對角線矩陣，這在處理線性代數問題和進行矩陣運算時非常有用。

最新問答