Построить полином Лагранжа из множества точек PYTHON

Question

Мне нужно нарисовать кривую, проходящую через точки (0,0), (20,10), (0,50).Поскольку есть две точки с одинаковыми координатами x, я решил параметризовать x и y параметром t, и, поскольку мне нужно уравнение этого многочлена (кривой), я решил использовать Лагранж.Ну, сюжет действительно странный, он печатает точки, но кривой нет:

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")

Пожалуйста, не могли бы вы показать мне, что не так с кодом?Заранее спасибо!

Answer 1

Вы должны переопределить переменные. Это работает

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()