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

yipeiwu_com6年前Python基础

背景:

实现用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设计】。

相关文章

Python实现大文件排序的方法

本文实例讲述了Python实现大文件排序的方法。分享给大家供大家参考。具体实现方法如下: import gzip import os from multiprocessing imp...

在pytorch 中计算精度、回归率、F1 score等指标的实例

pytorch中训练完网络后,需要对学习的结果进行测试。官网上例程用的方法统统都是正确率,使用的是torch.eq()这个函数。 但是为了更精细的评价结果,我们还需要计算其他各个指标。在...

python调用matlab的m自定义函数方法

项目信号处理和提取部分用到了matlab,需要应用到工程中方便研究。用具有万能粘合剂之称的“Python”。具体方法如下: 1.python中安装mlab 下载https://pypi...

pytorch模型预测结果与ndarray互转方式

预测结果转为numpy: logits=model(feature) #如果模型是跑在GPU上 result=logits.data.cpu().numpy() / logi...

python实现五子棋游戏

python实现五子棋游戏

本文实例为大家分享了python实现五子棋游戏的具体代码,供大家参考,具体内容如下 话不多说,直接上代码: 全部工程文件,在GitHub:五子棋 效果预览: #!/usr/bin/...