数学建模社区-数学中国

标题: 拉格朗日插值 python实现 [打印本页]
作者: 2744557306    时间: 2024-3-14 11:02
标题: 拉格朗日插值 python实现
让我们逐步解析和详细解释这段代码。$ D: @7 |: }0 G' p# a  k# q
1. 导入必要的库
  1. import numpy as np# k8 y( H; I! Z0 o8 Q
  2. ; C! j0 t# c( A' J, k
  3. from scipy.interpolate import lagrange
    5 Z- Q+ n# y8 \& R+ ]
  4. : [/ L3 s" X' y& D8 e0 k, P- W
  5. import matplotlib.pyplot as plt
复制代码
1.numpy: 一个强大的数学库，用于处理数组和矩阵等高级数学运算。
+ P1 P# E  Z% o/ t7 D# C0 I/ s2.scipy.interpolate.lagrange: 从 SciPy 库中导入的，用于执行拉格朗日多项式插值的函数。
# ^* A" {/ U( l2 X+ E0 S" Q6 M- `3.matplotlib.pyplot: 用于数据可视化的库，可以绘制图表和图形。, ?% q; a* p2 d
- a: O1 Y, C3 N/ i) v1 F3 `
2. 源数据
  1. x = np.array([4, 5, 6])
    9 m; v0 n2 V% e' d
  2. 9 g1 \* W( u, z
  3. y = np.array([10, 5.25, 1])
复制代码
这里定义了两个数组 x 和 y，代表了三个数据点的坐标：(4, 10), (5, 5.25), 和 (6, 1)。这些数据点将用于拉格朗日插值。
+ w* E; F$ {3 M( A0 M; j3. 拉格朗日插值
  1. poly = lagrange(x, y)
    6 a/ z9 ?. x- `5 [5 ?: L. O2 B$ W- R
  2. ) N1 ~0 \  ^8 C( r- ^
  3. res = poly(18)
复制代码
4.使用 lagrange(x, y) 创建一个拉格朗日插值多项式 poly。这个多项式是基于给定的数据点 x 和 y 计算出来的，它能够逼近这些点构成的曲线。
- i- ]) a1 i1 I  v; {- V7 q5.poly(18) 计算了在 x = 18 时的插值结果 res。这是通过将 x = 18 代入到拉格朗日多项式中得到的。5 e$ w: t4 n! D* s5 [% r# _5 w

* F+ A6 h( `8 b# ^5 r4 ^9 w* e4. 打印多项式和插值结果
  1. print('poly =\n', poly)
    3 v( k* J. f* v2 {
  2. . f9 t, Z$ g# x7 T+ O3 _
  3. print('res =', res)
复制代码
这里打印了插值多项式 poly 和在 x = 18 处的插值结果 res。
/ w: h  y3 i/ u! |" \, Y5. 数据可视化
  1. x1 = [i for i in range(-5, 20)]
    2 I5 ]3 y# g, m+ o8 V; O4 I

  2. & Z6 X" R7 ]. Z( q6 J. |  a# b
  3. y1 = []
    7 v' i) I" f' i
  4. . ]8 B9 t/ `) z" T+ z
  5. for each in x1:  c/ p* X# q8 G
  6. : h7 w- H( Z/ f1 ~& _+ ]9 P
  7.     y1.append(poly(each))% A* o0 \( v9 a( r

  8. " X& s$ b% s/ z! N1 Q

  9. : I) u" ]4 `1 U$ D
  10. + y1 m0 X0 D- v- X# E9 c5 `+ E! J
  11. plt.plot(x1, y1)1 t5 Y; }: q3 T# Z

  12. 3 [/ L' ]' p3 k" e% p
  13. plt.scatter(x, y)& ^" z. M" |  H/ _& A" F
  14. 4 c1 X% Z! Y7 I6 n
  15. plt.scatter(18, res)
复制代码
6.首先，创建了一个 x 值列表 x1，范围是从 -5 到 19。然后，对于 x1 中的每个 x 值，都计算对应的拉格朗日插值多项式的 y 值，并将这些 y 值存储在列表 y1 中。! e/ x0 L: R8 p+ U; u8 O( }
7.使用 plt.plot(x1, y1) 绘制插值多项式形成的曲线。. C8 F0 ^/ J/ i: e
8.使用 plt.scatter(x, y) 在图上标记原始数据点。7 b+ `1 g8 f5 D/ L5 h
9.使用 plt.scatter(18, res) 在图上标记 x = 18 处的插值结果点。
7 G" P5 o- b* l% g6 ^0 C% C5 h0 L9 t  a
这样，你就得到了一幅图形，显示了原始数据点、插值曲线，以及在 x = 18 处的插值点。这个过程展示了如何使用拉格朗日插值方法根据一组给定的数据点来估计函数在特定点上的值，并将结果可视化。
8 ^' p; P. [! z5 c+ V4 C1 B9 C1 v$ x, I- m
. d! |& I8 {, t7 K* _- v

09.lagrange_interpolation.py

570 Bytes, 下载次数: 0, 下载积分: 体力 -2 点

售价: 2 点体力  [记录]  [购买]





欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5