from sympy import *
x0=0
x1=25000
x2=50000
x3=75000
x4=100000
x5=125000

y0=9.8100
y1=9.7335
y2=9.6578
y3=9.5830
y4=9.5091
y5=9.4361

v=62500

x=Symbol("x")
L0=((x-x1)*(x-x2)*(x-x3)*(x-x4)*(x-x5))/((x0-x1)*(x0-x2)*(x0-x3)*(x0-x4)*(x0-x5))
L1=((x-x0)*(x-x2)*(x-x3)*(x-x4)*(x-x5))/((x1-x0)*(x1-x2)*(x1-x3)*(x1-x4)*(x1-x5))
L2=((x-x0)*(x-x1)*(x-x3)*(x-x4)*(x-x5))/((x2-x0)*(x2-x1)*(x2-x3)*(x2-x4)*(x2-x5))
L3=((x-x0)*(x-x1)*(x-x2)*(x-x4)*(x-x5))/((x3-x0)*(x3-x1)*(x3-x2)*(x3-x4)*(x3-x5))
L4=((x-x0)*(x-x1)*(x-x2)*(x-x3)*(x-x5))/((x4-x0)*(x4-x1)*(x4-x2)*(x4-x3)*(x4-x5))
L5=((x-x0)*(x-x1)*(x-x2)*(x-x3)*(x-x4))/((x5-x0)*(x5-x1)*(x5-x2)*(x5-x3)*(x5-x4))

P=y0*L0+y1*L1+y2*L2+y3*L3+y4*L4+y5*L5
P_ec=expand(P)

s=P_ec.subs(x,v)
print("La gravedad para una altura de",v,"es",s)

print("El polinomio interpolador es",P_ec)