python求解非线性多元方程_python用fsolve、leastsq对非线性方程组求解方法

python用fsolve、leastsq对非线性方程组求解总结

背景：

实现用python的optimize库的fsolve对非线性方程组进行求解。可以看到这一个问题实际上还是一个优化问题，也可以用之前拟合函数的leastsq求解。下面用这两个方法进行对比：

代码：

from scipy.optimize import fsolve,leastsq

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

]

x0 = [1,1,1]

result = fsolve(f,x0)

print("===================")

print()

print("求解函数名称:",fsolve.__name__)

print("解：",result)

print("各向量值：",f(result))

#拟合函数来求解

h = leastsq(f,x0)

print("===================")

print()

print("求解函数名称:",leastsq.__name__)

print("解：",h[0])

print("各向量的值：",f(h[0]))

结果：

===================

求解函数名称: fsolve

解： [-0.70622057 -0.6        -2.5       ]

各向量值： [0.0, -9.126033262418787e-14, 5.329070518200751e-15]

===================

求解函数名称: leastsq

解： [-0.70622057 -0.6        -2.5       ]

各向量的值： [0.0, -2.220446049250313e-16, 0.0]

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持码农之家。

python scipy求解非线性方程的方法(fsolve/root)

使用scipy.optimize模块的root和fsolve函数进行数值求解线性及非线性方程，下面直接贴上代码，代码很简单

from scipy.integrate import odeint

import numpy as np

import matplotlib.pyplot as plt

from scipy.optimize import root,fsolve

#plt.rc('text', usetex=True) #使用latex

## 使用scipy.optimize模块的root和fsolve函数进行数值求解方程

## 1、求解f(x)=2*sin(x)-x+1

rangex1 = np.linspace(-2,8)

rangey1_1,rangey1_2 = 2*np.sin(rangex1),rangex1-1

plt.figure(1)

plt.plot(rangex1,rangey1_1,'r',rangex1,rangey1_2,'b--')

plt.title('$2sin(x)$ and $x-1$')

def f1(x):

return np.sin(x)*2-x+1

sol1_root = root(f1,[2])

sol1_fsolve = fsolve(f1,[2])

plt.scatter(sol1_fsolve,2*np.sin(sol1_fsolve),linewidths=9)

plt.show()

## 2、求解线性方程组{3X1+2X2=3;X1-2X2=5}

def f2(x):

return np.array([3*x[0]+2*x[1]-3,x[0]-2*x[1]-5])

sol2_root = root(f2,[0,0])

sol2_fsolve = fsolve(f2,[0,0])

print(sol2_fsolve) # [2. -1.5]

a = np.array([[3,2],[1,-2]])

b = np.array([3,5])

x = np.linalg.solve(a,b)

print(x) # [2. -1.5]

## 3、求解非线性方程组

def f3(x):

return np.array([2*x[0]**2+3*x[1]-3*x[2]**3-7,

x[0]+4*x[1]**2+8*x[2]-10,

x[0]-2*x[1]**3-2*x[2]**2+1])

sol3_root = root(f3,[0,0,0])

sol3_fsolve = fsolve(f3,[0,0,0])

print(sol3_fsolve)

## 4、非线性方程

def f4(x):

return np.array(np.sin(2*x-np.pi)*np.exp(-x/5)-np.sin(x))

init_guess =np.array([[0],[3],[6],[9]])

sol4_root = root(f4,init_guess)

sol4_fsolve = fsolve(f4,init_guess)

print(sol4_fsolve)

t = np.linspace(-2,12,2000)

y1 = np.sin(2*t-np.pi)*np.exp(-t/5)

y2 = np.sin(t)

plt.figure(2)

a , = plt.plot(t,y1,label='$sin(2x-\pi)e^{-x/5}$')

b , = plt.plot(t,y2,label='$sin(x)$')

plt.scatter(sol4_fsolve,np.sin(sol4_fsolve),linewidths=8)

plt.title('$sin(2x-\pi)e^{-x/5}$ and $sin(x)$')

plt.legend()

以上这篇python scipy求解非线性方程的方法(fsolve/root)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持码农之家。

以上就是本次给大家分享的关于java的全部知识点内容总结，大家还可以在下方相关文章里找到相关文章进一步学习，感谢大家的阅读和支持。

您可能感兴趣的文章: