非线性方程组求解

君不见长松卧壑困风霜,时来屹立扶明堂。这篇文章主要讲述非线性方程组求解相关的知识,希望能为你提供帮助。
非线性方程组求解
fsolve() 可以对非线性方程组进行求解,它的基本调用形式为fsolve(func,x0)。其中 func是 计算方程组误差的函数,它的参数x 是一个数组,其值为方程组的一组可能的解。func返回将 x 代入方程组之后得到的每个方程的误差,x0为未知数的一组初始值。假设要对下而的方程组进行求解:

f1(u1 ,u2, u3) =0,f2(u1 , u2, u3) =0,f3(u1 , u2, u3) =0

那么func函数可以定义如下:
def func(x): u1,u2,u3 = x return[f1(u1,u2,u3),f2(u1,u2,u3),f3(u1,u2,u3)]

下而我们看一个对下列方程组求解的例子:
非线性方程组求解

文章图片

from math import sin,cos from scipy import optimizedef f(x): x0,x1,x2 = x.tolist() return [5*x1+3,4*x0*x0-2*sin(x1*x2),x1*x2-1.5]# f 计算方程组的误差,[1,1,1]是未知数的初始值 result = optimize.fsolve(f,[1,1,1]) print(result) # x0,x1,x2的值 print(f(result)) # 方程组的误差

[-0.70622057 -0.6-2.5] [0.0, -9.126033262418787e-14, 5.329070518200751e-15]

  • f()是计算方程组的误差的函数,X参数是一绀可能的解。fsolve()在调用 f() 时,传递给f() 的参数是一个数组。
  • 先调用数组的tolist()方法,将其转换为python的标准浮点数列表,然后调用math模块中的阑数进行运算。因为在进行单个数值的运算时,标准浮点类型比NumPy的 浮点类型要快许多,所以把数值都转换成标准浮点数类型,能缩短一些计算时间。
  • 调用fsolve() 时,传递计算误差的函数f()以及未知数的初始值。
? 在对方程组进行求解吋,fsolve()会自动计算方程组在某点对各个未知数变量 的偏导数,这些偏导数组成一个二维数组,数学上称之为雅可比矩阵。如果方程组中的未知数很多,而与每个方程有关联的未知数较少,即雅可比矩阵比较稀疏吋,将计算雅可比矩阵的函数作为参数传递给 fsolve(), 这能大幅度提高运算速度。笔者在一个模拟计算的程序中需要求解有50个未知数的非线性方程组。每个方程平均与6 个未知数相关,通过传递计箅雅可比矩阵的函数使fsolve()的计算速度提高了 4 倍。
雅可比矩阵
雅可比矩阵是一阶偏导数以一定方式排列的矩阵,它给出了可微分方程与给定点的最优线 性逼近,因此类似于多元函数的导数。例如前面的函数fl、f2、f3 和未知数u1、u2、u3 的雅可比矩阵如下:
非线性方程组求解

文章图片


? 下面使用雅可比矩阵对方程组进行求解。计算雅可比矩阵的函数j()和f()— 样,其 x 参数是未知数的一组值,它计算非线性方程组在x 处的雅可比矩阵。通过fprime () 参数将 j()传递 给 fsolve() 。由于本例中的未知数很少,因此计算雅可比矩阵并不能显著地提高计算速度。
def f(x): x0,x1,x2 = x.tolist() return [5*x1+3,4*x0*x0-2*sin(x1*x2),x1*x2-1.5]def j(x): x0,x1,x2 =x.tolist() return [[0,5,0], [8*x0,-2*x2*cos(x1*x2),-2*x1*cos(x1*x2)], [0,x2,x1]]result = optimize.fsolve(f,[1,1,1],fprime=j) print(result) print(f(result))

[-0.70622057 -0.6-2.5] [0.0, -9.126033262418787e-14, 5.329070518200751e-15]

【非线性方程组求解】摘自《python科学计算》

    推荐阅读