Python Sympy:解方程利器

2023年 7月 14日 28.2k 0

用程序来解决数学问题是非常普遍的,将数学的定理或公式封装成程序中的函数,只要传入相应的参数,就能让计算机帮我们计算出最终的结果。

不过,今天介绍的这个库:Sympy,它的最大特点是让我们可以用做数学题的思考方式来写程序。

1. 变量和表达式

用程序实现数学的算法,会根据程序语言本身的特点来实现算法,不会像解数学问题那样一步步推导。

所以,虽然可以用程序解决很多的数学问题,但是最后将代码展现出来时,数学专业的朋友也许很难看懂。Sympy 让我们可以用数学专业的朋友熟悉的方式来写程序。

1.1. 变量

Sympy 的变量是一个数学符号,和我们平时理解的程序中的变量不太一样。

from sympy import Symbol

x = Symbol("x")
x.name
#输出结果:
x

x是程序中的变量,它代表的数学符号也是x

这么写可以有点迷惑,改成下面这样可能好理解一些:

a = Symbol("x")
a.name
#输出结果:
x

a是程序中的变量,它代表的数学符号是x

1.2. 表达式

定义了数学符号之后,就可以自由的写各种数学表达式了。比如:

x = Symbol("x")
expr = x*x + 1
print(expr)
#输出结果:
x**2 + 1

y = Symbol("y")
expr = x*x - (y-1)*(y-1)
print(expr)
#输出结果:
x**2 - (y - 1)**2

程序中的表达式,Sympy会转换成数学的形式,以用于数学的计算。

1.3. 因式分解和展开

因式分解是初中常做的数学题目之一,用Sympy解决这类问题非常简单。

from sympy import Symbol, factor, expand

x = Symbol("x")
y = Symbol("y")

#分解
expr = (x-2)*(x-2) - (y-3)*(y-3)
print(factor(expr))
#输出结果
(x - y + 1)*(x + y - 5)

#展开
expr = (x - y + 1)*(x + y - 5)
expr = expand(expr)
print(expr)
#输出结果
x**2 - 4*x - y**2 + 6*y - 5

1.4. 表达式求解

Sympy让我们可以在计算时保留数学符号(比如x,y),也可以代入实际的数字,最终求出表达式的值。

#沿用上面的表达式
print(expr)
#输出结果
x**2 - 4*x - y**2 + 6*y - 5

result = expr.subs({x:2, y:3})
print(result)
#输出结果
0

甚至可以用数学符号来替换,比如假设 x = y - 1 ,替换上面的表达式。然后再求值:

expr = expr.subs({x: y - 1})
print(expr)
#输出结果
-y**2 + 2*y + (y - 1)**2 - 1

result = expr.subs({y: 3})
print(result))
#输出结果
0

先设置 x = y - 1,然后设 y = 3,和直接设置 x = 2, y = 3 是一样的。所以最后计算结果也是0

1.5. 表达式计算

Sympy 还可以支持表达式之间的计算。

#表达式之间加法
expr1 = x * y + y * y
expr2 = x * x + x * y

expr = expr1 + expr2
print(expr)
#输出结果
x**2 + 2*x*y + y**2

#因式分解
print(factor(expr))
#输出结果
(x + y)**2

除了加法,表达式之间也可以相减,相乘,相除也是可以的。

2. 解方程

用程序解方程问题很复杂,一般得用到牛顿迭代法之类的迭代算法。用Sympy解方程,只要一行代码,:)

2.1. 一元方程

Sympysolve函数封装好了方程解法。

from sympy import solve

expr = x*x - 5*x + 6
solve(expr)
#输出结果
[2, 3]

solve(expr, dict=True)
#输出结果(字典形式)
[{x: 2}, {x: 3}]

2.2. 二元方程组

二元方程组也一样:

expr1 = 3*x + 4*y - 34
expr2 = 2*x + 3*y - 56

solve((expr1, expr2), dict=True)
#输出结果(字典形式)
[{x: -122, y: 100}]

3. 函数图像

最后,介绍下Sympy绘制函数图像的功能。绘制函数的图像能够让我们更加直观的看到函数的变化趋势,对于理解题目非常有帮助。

使用 Sympy 绘制图像非常方便,只要传入数学表达式即可。

from sympy import plot

plot(2*x + 3)

image.png

设置 X 轴的范围(-5~5):

from sympy import plot

plot(2*x + 3, (x, -5, 5))

image.png

绘制多个函数的图像:

plot(2*x + 3, 5*x + 2, (x, -5, 5), legend=True)

image.png

4. 总结回顾

本篇主要介绍 Sympy 在基本的代数运算方面的应用,Sympy 最大的特点是让我们可以用数学专业的思维方式来写代码。

它的代码与数学的联系非常紧密,数学专业的朋友如果刚接触编程,用这个库来写代码一定会非常亲切。

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论