Python | sympy.eye() 方法介绍

sympy 是 Python 中的一个强大的符号计算库，eye() 方法是其中的一个特别有用的函数。

eye() 方法

sympy.eye(n, m=None, k=0, dtype=<class 'int'>)

参数说明

• n: 输出矩阵的行数
• m: 输出矩阵的列数，如果不提供，则默认为 n
• k: 对角线的位置。k = 0 表示主对角线，k > 0 表示在主对角线上方的对角线，k < 0 表示在主对角线下方的对角线
• dtype: 矩阵元素的数据类型，默认为 int

返回值

一个 n x m 的对角线矩阵，对角线上的元素为 1，其余元素为 0。

例子

import sympy

# 输出一个 3x3 的对角矩阵
a = sympy.eye(3)
print(a)

输出结果如下:

Matrix([[1, 0, 0], [0, 1, 0], [0, 0, 1]])

总结

sympy.eye() 方法提供了一个便捷的方法来创建对角矩阵，这在一些数学计算中常常会用到。有了 sympy，数学计算会变得更加轻松。