python eye函数的语法结构解析

numpy.eye 是 NumPy 库中的一个函数，用于创建一个对角线上元素为 1，其余元素为 0 的二维数组（也称为单位矩阵）

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

参数说明：

• N：输出矩阵的行数。必需参数。
• M：输出矩阵的列数。如果未指定，默认值为 N，即创建一个方阵。
• k：指定对角线的位置。正值表示上对角线，负值表示下对角线，0 表示主对角线。默认值为 0。
• dtype：输出矩阵的数据类型。默认值为 float。
• order：输出矩阵的内存布局。可以是 'C'（按行优先存储）或 'F'（按列优先存储）。默认值为 'C'。
• like：用于指定输出矩阵的类型和属性。如果提供了此参数，将根据 like 的类型和属性创建输出矩阵。

示例：

import numpy as np

# 创建一个 3x3 的单位矩阵
identity_matrix = np.eye(3)
print(identity_matrix)

# 创建一个 4x4 的单位矩阵，其中对角线上的元素为 2
scaled_identity_matrix = np.eye(4, dtype=int) * 2
print(scaled_identity_matrix)

# 创建一个 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.]]