从一组点PYTHON绘制拉格朗日多项式

时间:2018-06-03 16:18:12

标签: python python-3.x numpy plot scipy

我需要绘制一条穿过点(0,0),(20,10),(0,50)的曲线。由于有两个点具有相同的x坐标,我决定用参数t对x和y进行参数化,因为我需要这个多项式(曲线)的方程式,所以我决定使用拉格朗日。 嗯,情节真的很奇怪,它会打印点,但没有曲线:weird plot

x_parabola = np.array([0, 20, 0]) # array for x
y_parabola = np.array([0, 10, 50]) # array for y
plt.figure()
u = plt.plot(x_parabola,y_parabola,'ro') # plot the points
t = np.linspace(0, 1, len(x_parabola)) # parameter t to parametrize x and y
pxLagrange = scipy.interpolate.lagrange(t, x) # X(T)
pyLagrange = scipy.interpolate.lagrange(t, y) # Y(T)
n = 100
ts = np.linspace(t[0],t[-1],n)
xLagrange = pxLagrange(ts) # lagrange x coordinates
yLagrange = pyLagrange(ts) # lagrange y coordinates
plt.plot(xLagrange, yLagrange,'b-',label = "Polynomial")

拜托,您能告诉我代码有什么问题吗?提前谢谢!

1 个答案:

答案 0 :(得分:2)

你必须重新定义变量。这可行

import numpy as np
import scipy.interpolate
import matplotlib.pyplot as plt

x_parabola = np.array([0, 20, 0]) # array for x
y_parabola = np.array([0, 10, 50]) # array for y
plt.figure()
u = plt.plot(x_parabola,y_parabola,'ro') # plot the points
t = np.linspace(0, 1, len(x_parabola)) # parameter t to parametrize x and y
pxLagrange = scipy.interpolate.lagrange(t, x_parabola) # X(T)
pyLagrange = scipy.interpolate.lagrange(t, y_parabola) # Y(T)
n = 100
ts = np.linspace(t[0],t[-1],n)
xLagrange = pxLagrange(ts) # lagrange x coordinates
yLagrange = pyLagrange(ts) # lagrange y coordinates
plt.plot(xLagrange, yLagrange,'b-',label = "Polynomial")
plt.show()