python 实现(拟)牛顿法解非线性方程组

最近需要解一个非线性方程组,发现网上资料很少,我自己写了一个,供大家参考。
例题参考
https://wenku.baidu.com/view/a87f1e6a58fafab069dc0260.html?from=search

import math
import numpy as np  
from numpy import * 
import matplotlib.pyplot as plt

def Fun(x,num):                       #方程组在这里,三个变量分别是x的三个分量,num是未知数个数,这里是3,f是三个方程组 
    i = num
    f = np.zeros((i),dtype=float)
    f[0] = x[0]*x[1]-x[2]*x[2]-1.
    f[1] = x[0]*x[1]*x[2]+x[1]*x[1]-x[0]*x[0]-2.
    f[2] = math.exp(x[0])+x[2]-math.exp(x[1])-3.
    return f

def dfun(x,num):                         #计算雅可比矩阵的逆矩阵
    df = np.zeros((num,num),dtype=float)
    dx = 0.00001                           # 
    x1 = np.copy(x)
    for i in range(0,num):              # 求导数,i是列,j是行
        for j in range(0,num):
            x1 = np.copy(x)
            x1[j] = x1[j]+dx           #x+dx
            df[i,j] = (Fun(x1,num)[i]-Fun(x,num)[i])/dx   #f(x+dx)-f(x)/dx
    df_1 = np.linalg.inv(df)                              #计算逆矩阵
    return df_1

def Newton(x,num):
    x1 = np.copy(x)
    i = 0
    delta = np.copy(x)
#    dfun0=dfun(x,num)          #也可以使用简化牛顿法
    while(np.sum(abs(delta)) > 1.e-8 and i < 20):  #控制循环次数
        x1 = x-dot(dfun(x,num),Fun(x,num))  #公式
        delta = x1-x                     #比较x的变化
        x = x1
        i = i+1
        print(x)
    return x

num =3                      # 方程未知数的个数
x = np.ones((num),dtype=float)#初始值
a = Newton(x,num)
print(a)
可以使用Python中的牛顿法来求解非线性方程组牛顿法是一种迭代的方法,通过逐步逼近函数的根。在Python中,可以使用scipy.optimize库中的fsolve函数来实现牛顿法解非线性方程组。 首先,需要定义一个函数,该函数返回一个数组,数组的元素是线性方程组的各个方程的值。然后,使用fsolve函数传入定义的函数和一个初始的猜测解来求解非线性方程组。fsolve函数会返回一个数组,数组的元素是线性方程组的解。 下面是一个使用牛顿法解非线性方程组的示例代码: ```python from scipy.optimize import fsolve # 定义线性方程组 def equations(x): # x和x是未知数 y1 = 2*x - x - 2 y2 = x # 初始猜测解 result = fsolve(equations, x0) print(result) ``` 在上述代码中,我们定义了一个线性方程组,包含两个方程。然后,使用fsolve函数传入定义的函数和一个初始的猜测解来求解方程组。最后,打印出求解结果。 请注意,对于复杂的线性方程组牛顿法的收敛性可能<span class="em">1</span> #### 引用[.reference_title] - *1* [基于Python利用Newton-Raphson方法求解非线性方程组](https://blog.csdn.net/weixin_39964552/article/details/84331937)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值