eye 函数是 NumPy 库中用于创建单位矩阵(identity matrix)的一个函数
以下是 eye 函数与其他矩阵函数的比较:
numpy.eye(N, M=None, k=0, dtype=float):创建一个 NxM 的单位矩阵,其中 N 是行数,M 是列数(默认为 N),k 是对角线索引(0 表示主对角线,正数表示上对角线,负数表示下对角线)。dtype 参数指定矩阵元素的数据类型。import numpy as np I = np.eye(3) print(I) 输出:
[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]] numpy.zeros((N, M), dtype=float):创建一个 NxM 的全零矩阵。Z = np.zeros((3, 3)) print(Z) 输出:
[[0. 0. 0.] [0. 0. 0.] [0. 0. 0.]] numpy.ones((N, M), dtype=float):创建一个 NxM 的全一矩阵。O = np.ones((3, 3)) print(O) 输出:
[[1. 1. 1.] [1. 1. 1.] [1. 1. 1.]] numpy.diag(v, k=0):从给定向量 v 创建一个对角矩阵,k 参数表示对角线索引。D = np.diag([1, 2, 3]) print(D) 输出:
[[1 0 0] [0 2 0] [0 0 3]] 总结:eye 函数专门用于创建单位矩阵,而其他矩阵函数可以创建全零、全一或自定义对角线的矩阵。在需要单位矩阵时,使用 eye 函数更直观和高效。