python eye函數(shù)的語法結(jié)構(gòu)解析

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

numpy.eye(N, M=None, k=0, dtype=float, order='C', *, like=None)

參數(shù)說明：

• N：輸出矩陣的行數(shù)。必需參數(shù)。
• M：輸出矩陣的列數(shù)。如果未指定，默認(rèn)值為 N，即創(chuàng)建一個方陣。
• k：指定對角線的位置。正值表示上對角線，負(fù)值表示下對角線，0 表示主對角線。默認(rèn)值為 0。
• dtype：輸出矩陣的數(shù)據(jù)類型。默認(rèn)值為 float。
• order：輸出矩陣的內(nèi)存布局?？梢允?'C'（按行優(yōu)先存儲）或 'F'（按列優(yōu)先存儲）。默認(rèn)值為 'C'。
• like：用于指定輸出矩陣的類型和屬性。如果提供了此參數(shù)，將根據(jù) like 的類型和屬性創(chuàng)建輸出矩陣。

示例：

import numpy as np

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

# 創(chuàng)建一個 4x4 的單位矩陣，其中對角線上的元素為 2
scaled_identity_matrix = np.eye(4, dtype=int) * 2
print(scaled_identity_matrix)

# 創(chuàng)建一個 4x4 的矩陣，其中第 2 條對角線上的元素為 1
diagonal_matrix = np.eye(4, k=1)
print(diagonal_matrix)

輸出：

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

[[2 0 0 0]
 [0 2 0 0]
 [0 0 2 0]
 [0 0 0 2]]

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