sympy库用着稍微有点问题,解简单的方程还行,稍微复杂点的结果就不太行,要么打印出了一堆奇怪的东西,要么会运行失败。自己写了一个解方程的函数,原理是穷尽搜索解的值,代码如下。left和right是把等式两边变成函数,x_range是解的搜索范围,step是搜索步长,error是允许的误差。
import numpy as np
def solve_equation(left, right, x_range:list, step:float, error:float):
result = []
all_x = np.arange(x_range[0], x_range[1] + step, step)
for x in all_x:
left_x = left(x)
right_x = right(x)
if abs(left_x - right_x) < error:
result.append(x)
return result
比如要找出sin和cos在[0,1]区间内的交点,即解sin(x)=cos(x):
print(solve_equation(np.sin, np.cos, [0, 1], 0.00001, 0.00001))
打印结果是0.7854000000000001,和真实值0.7853981633974483还是比较接近的。