eye函数在python中的参数设置

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

numpy.eye(N, M=None, k=0, dtype=<class 'float'>) 

参数说明：

• N：输出矩阵的行数。必需参数。
• M：输出矩阵的列数。可选参数，默认值为 N，即创建一个方阵。
• k：指定对角线的位置。可选参数，默认值为 0，表示主对角线。正值表示上对角线，负值表示下对角线。
• dtype：输出矩阵的数据类型。可选参数，默认值为 float 类型。

示例：

import numpy as np  # 创建一个 3x3 的单位矩阵 identity_matrix = np.eye(3) print("3x3 identity matrix:\n", identity_matrix)  # 创建一个 4x4 的单位矩阵 identity_matrix_4x4 = np.eye(4) print("\n4x4 identity matrix:\n", identity_matrix_4x4)  # 创建一个 3x4 的矩阵，其中第 2 条对角线上的元素为 1 matrix_3x4 = np.eye(3, 4, k=1) print("\n3x4 matrix with 2nd diagonal ones:\n", matrix_3x4)  # 创建一个 4x4 的整数单位矩阵 integer_identity_matrix = np.eye(4, dtype=int) print("\n4x4 integer identity matrix:\n", integer_identity_matrix) 

输出结果：

3x3 identity matrix:  [[1. 0. 0.]  [0. 1. 0.]  [0. 0. 1.]]  4x4 identity matrix:  [[1. 0. 0. 0.]  [0. 1. 0. 0.]  [0. 0. 1. 0.]  [0. 0. 0. 1.]]  3x4 matrix with 2nd diagonal ones:  [[0. 1. 0. 0.]  [0. 0. 1. 0.]  [0. 0. 0. 1.]]  4x4 integer identity matrix:  [[1 0 0 0]  [0 1 0 0]  [0 0 1 0]  [0 0 0 1]]