非线性方程组求解

君不见长松卧壑困风霜，时来屹立扶明堂。这篇文章主要讲述非线性方程组求解相关的知识，希望能为你提供帮助。
非线性方程组求解
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))