eye函数在python中的实际应用

eye 函数在 Python 中主要用于创建一个对角线上元素为 1，其余元素为 0 的二维数组

以下是 eye 函数在 Python 中的一些实际应用：

1. 创建单位矩阵：

import numpy as np

identity_matrix = np.eye(3)
print(identity_matrix)

输出结果：

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

2. 创建自定义对角线矩阵：

import numpy as np

custom_diagonal_matrix = np.eye(3, k=1)
print(custom_diagonal_matrix)

输出结果：

[[0. 1. 0.]
 [0. 0. 1.]
 [0. 0. 0.]]

3. 创建具有指定数据类型和对角线元素的矩阵：

import numpy as np

custom_dtype_matrix = np.eye(3, dtype=int)
print(custom_dtype_matrix)

输出结果：

[[1 0 0]
 [0 1 0]
 [0 0 1]]

4. 使用 eye 函数进行矩阵运算：

import numpy as np

A = np.array([[1, 2], [3, 4]])
I = np.eye(2)
B = A + I
print(B)

输出结果：

[[2. 2.]
 [3. 5.]]

总之，eye 函数在 Python 中可以用于创建和操作对角线矩阵，这在处理线性代数问题和进行矩阵运算时非常有用。