阅读量:0
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.]] 