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]]