Python 求解非线性方程
如下方程,求解x1,x2,x3:
5*x1 + 3 = 0
4*x0*x0 - 2*sin(x1*x2) = 0
x1*x2 - 1.5 = 0
from scipy.optimize import fsolve
from math import sin,cos
def f(x):
x0 = float(x[0])
x1 = float(x[1])
x2 = float(x[2])
return [
5*x1+3,
4*x0*x0 - 2*sin(x1*x2),
x1*x2 - 1.5
]
result = fsolve(f, [1,1,1])
print result
print f(result)
输出:
[-0.70622057 -0.6 -2.5 ]
[0.0, -9.1260332624187868e-14, 5.3290705182007514e-15]
fsolve(f, [1,1,1]) 参数中的数组,为一组可能解,作为初始值,进行迭代,得到最终解。使用scipy时,必须要有安装numpy。
帮助来源:scipy 官方解释
【http://old.sebug.net/paper/books/scipydoc/scipy_intro.html#id3】