画出拉格朗日插值函数图像python

下面是使用Python代码绘制拉格朗日插值函数的图像：

import numpy as np
import matplotlib.pyplot as plt

def lagrange_interpolation(x, y, x_test):
    N = len(x)
    L = np.zeros(N)
    for i in range(N):
        xi = x[i]
        yi = y[i]
        l = 1
        for j in range(N):
            if i != j:
                xj = x[j]
                l *= (x_test - xj) / (xi - xj)
        L[i] = yi * l
    return np.sum(L)

# 构造测试数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([0.5, 0.4, 0.3, 0.6, 0.8])

# 生成插值数据
x_test = np.linspace(0.5, 5.5, 100)
y_test = np.zeros_like(x_test)
for i in range(len(x_test)):
    y_test[i] = lagrange_interpolation(x, y, x_test[i])

# 绘制图像
plt.plot(x, y, 'o', label='data points')
plt.plot(x_test, y_test, label='interpolation')
plt.legend(loc='best')
plt.show()

此代码将生成一个拉格朗日插值函数的图像，其中包括原始数据点和插值函数的曲线。