插值方法 - Lagrange插值多项式

Lagrange插值多项式代码：

# -*- coding: utf-8 -*-
"""
Created on Wed Mar 25 15:43:42 2020

@author: 35035
"""


import numpy as np

# Lagrange插值多项式：
def Lagrange(x, y, xi):
    '''x、y是array，xi是一个值，函数返回结果ans'''
    n = len(x)
    m = len(y)
    if n != m:
        print("Error!")
        return None
    ans = 0
    for i in range(n):
        # 选定第i项循环计算
        temp = 1
        for j in range(n):
            if i == j:
                continue
            temp *= (xi - x[j]) / (x[i] - x[j])
        ans += temp * y[i]
    return ans

# 当对多个值使用Lagrange插值时，使用map()建立映射:
# Iterator = map(Lagrange, Iterable)

# 用单个值测试函数：计算根号115

# 线性插值：
# 取x0 = 100, y0 = 10; x1 = 121, y1 = 11
# 抛物线插值：
# 多取一点 x2 = 144, y2 = 12
y = np.array((10, 11))    # 12
x = np.array((100, 121))  # 144
ans = Lagrange(x, y, 115)
print(ans)   
# 线性插值结果：10.413   抛物线插值结果：10.7228
# 测试成功！

还想打一下公式的，搞了几小时结果对结果都不满意，果然在这个编辑器数学公式与代码不可兼得......