阅读量:0
eye
函数在 Python 中主要用于创建一个对角线上元素为 1,其余元素为 0 的二维数组
以下是 eye
函数在 Python 中的一些实际应用:
- 创建单位矩阵:
import numpy as np identity_matrix = np.eye(3) print(identity_matrix)
输出结果:
[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]
- 创建自定义对角线矩阵:
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.]]
- 创建具有指定数据类型和对角线元素的矩阵:
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]]
- 使用
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 中可以用于创建和操作对角线矩阵,这在处理线性代数问题和进行矩阵运算时非常有用。