欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

拉格朗日插值--等距节点Python实现并计算误差

程序员文章站 2022-06-07 09:57:01
...

目标函数

y=11+x2

条件

通过拉格朗日进行插值但是通过所给的节点的不同,会导致插值的效果也不同。

下面方法采用的是用等距节点来实现插值效果。

插值节点分别是

-5,-4,-3,-2,-1,0,1,2,3,4,5

插值效果图

拉格朗日插值--等距节点Python实现并计算误差

龙格显现很严重,不过大致吻合。

代码

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


def f(x):
    return 1 / (1 + x ** 2)


def draw(L):
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.rcParams['axes.unicode_minus'] = False
    x = np.linspace(-5, 5, 100)
    y = f(x)
    Ly = []
    for xx in x:
        Ly.append(L.subs(n, xx))
    plt.plot(x, y, label='原函数')
    plt.plot(x, Ly, label='Lagrange插值函数')
    plt.xlabel('x')
    plt.ylabel('y')
    plt.legend()

    plt.savefig('1.png')
    plt.show()


def lossCal(L):
    x = np.linspace(-5, 5, 101)
    y = f(x)
    Ly = []
    for xx in x:
        Ly.append(L.subs(n, xx))
    Ly = np.array(Ly)
    temp = Ly - y
    temp = abs(temp)
    print(temp.mean())


if __name__ == '__main__':
    x = np.array(range(11)) - 5
    y = f(x)

    n, m = symbols('n m')
    init_printing(use_unicode=True)
    L = 0
    for k in range(11):
        temp = y[k]
        for i in range(11):
            if i != k:
                temp *= (n - x[i]) / (x[k] - x[i])
        L += temp
    lossCal(L)
    draw(L)