插值方法 - Lagrange插值多项式

Lagrange插值多项式代码：

 1 # -*- coding: utf-8 -*-
 2 """
 3 Created on Wed Mar 25 15:43:42 2020
 4
 5 @author: 35035
 6 """
 7
 8
 9 import numpy as np
10
11 # Lagrange插值多项式：
12 def Lagrange(x, y, xi):
13     '''x、y是array，xi是一个值，函数返回结果ans'''
14     n = len(x)
15     m = len(y)
16     if n != m:
17         print("Error!")
18         return None
19     ans = 0
20     for i in range(n):
21         # 选定第i项循环计算
22         temp = 1
23         for j in range(n):
24             if i == j:
25                 continue
26             temp *= (xi - x[j]) / (x[i] - x[j])
27         ans += temp * y[i]
28     return ans
29
30 # 当对多个值使用Lagrange插值时，使用map()建立映射:
31 # Iterator = map(Lagrange, Iterable)
32
33 # 用单个值测试函数：计算根号115
34
35 # 线性插值：
36 # 取x0 = 100, y0 = 10; x1 = 121, y1 = 11
37 # 抛物线插值：
38 # 多取一点 x2 = 144, y2 = 12
39 y = np.array((10, 11))    # 12
40 x = np.array((100, 121))  # 144
41 ans = Lagrange(x, y, 115)
42 print(ans)
43 # 线性插值结果：10.413   抛物线插值结果：10.7228
44 # 测试成功！

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