非线性方程组求解

非线性方程组求解

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 optimize

def 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 的雅可比矩阵如下:
\left[\begin{array}{lll}{\frac{\partial f 1}{\partial u 1}} & {\frac{\partial f 1}{\partial u 2}} & {\frac{\partial f 1}{\partial u}} \\ {\frac{\partial f 2}{\partial u 1}} & {\frac{\partial f 2}{\partial u 2}} & {\frac{\partial f 2}{\partial u 3}} \\ {\frac{\partial f 3}{\partial u 1}} & {\frac{\partial f 3}{\partial u 2}} & {\frac{\partial f 3}{\partial u 3}}\end{array}\right]
​ 下面使用雅可比矩阵对方程组进行求解。计算雅可比矩阵的函数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科学计算》

你可能感兴趣的:(非线性方程组求解)