帮我用Python实现一个拉格朗日插值算法

下面是一个 Python 实现的拉格朗日插值算法示例：

def lagrange_interpolation(x, y, xi):
    """
    使用拉格朗日插值算法计算函数在给定点的近似值
    :param x: 数据点的 x 坐标
    :param y: 数据点的 y 坐标
    :param xi: 要求函数值的点的 x 坐标
    :return: 函数在 xi 处的近似值
    """
    n = len(x)
    yi = 0.0
    for i in range(n):
        # 计算拉格朗日基函数的值
        L = 1.0
        for j in range(n):
            if j != i:
                L *= (xi - x[j]) / (x[i] - x[j])
        # 计算插值函数的值
        yi += L * y[i]
    return yi

使用示例：

x = [0, 1, 2, 3, 4, 5]
y = [0, 1, 4, 9, 16, 25]
xi = 2.5
yi = lagrange_interpolation(x, y, xi)
print(yi)  # 输出: 6.25

在这个示例中，我们使用了拉格朗日插值算法来计算函数 `y = x^2` 在 `x = 2.5` 处的函数值。我们使用了 6 个数据点，分别是 `(0,0)`, `(1,1)`, `(2,4)`, `(3,9)`, `(4,16)`, `(5,25)`。我们使用 `lagrange_interpolation` 函数计算出函数在 `x = 2.5` 处的近似值为 `6.25`。