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
函数更直观和高效。