eye 函数在 Python 中主要用于创建一个对角线上元素为 1,其余元素为 0 的二维数组
以下是 eye 函数在 Python 中的一些实际应用:
import numpy as npidentity_matrix = np.eye(3)print(identity_matrix)输出结果:
[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]创建自定义对角线矩阵:import numpy as npcustom_diagonal_matrix = np.eye(3, k=1)print(custom_diagonal_matrix)输出结果:
[[0. 1. 0.] [0. 0. 1.] [0. 0. 0.]]创建具有指定数据类型和对角线元素的矩阵:import numpy as npcustom_dtype_matrix = np.eye(3, dtype=int)print(custom_dtype_matrix)输出结果:
[[1 0 0] [0 1 0] [0 0 1]]使用 eye 函数进行矩阵运算:import numpy as npA = np.array([[1, 2], [3, 4]])I = np.eye(2)B = A + Iprint(B)输出结果:
[[2. 2.] [3. 5.]]总之,eye 函数在 Python 中可以用于创建和操作对角线矩阵,这在处理线性代数问题和进行矩阵运算时非常有用。


