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)
  • 32
    点赞
  • 81
    收藏
    觉得还不错? 一键收藏
  • 16
    评论
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值