Python - Necesito ayuda con tarea sobre un cohete

y aquí los datos del ejercicio ya resuelto:

1) Desde el despegue hasta que se apagan los motores:

Datos:

m = 7500 kg

a = 2,25 m / s^2

h final = 525 m

respuesta: Vo = 0

Tipo de movimiento: uniformemente acelerado

Ecuaciones:

altura = distancia recorrida = d = [1/2] a * t^2

V = a*t

V = 2ad

=> V^2 final = 2 * 2.25 m/s^2 * 525m = 2,362.5 m^2 / s^ => V final = 48.6 m/s

respuesta: t = Vf / a = [48.6 m/s ] / [2.25 m/s^2] = 21.6 s.

2) Desde que se apagaron los motores

Datos:

Vo = 48.6 m /s

tiempo para alcanzar la altura máxima:

Vf = Vo - g*t => t = [Vo - Vf] / g

=> t = [48.6 m/s] / 9.8 m/s^2 = 4.96 s

Altura alcanzada,h:

h = altura inicial + distancia recorrida hacia arriba

h = 585 m + Vo*t - a*t^2 / 2 = 585m + 48.6m/s*4.96s - 9.8m/s^2 * (4.96s)^2 / 2

respuesta: h = 705.5 m

De allí en adelante el cohete empieza a caer, hasta llegar al suelo, por lo que las ecuaciones que rigen el movimiento son:

h = 705.5 m - a*t^2 / 2 = 705m - 4.9 t^2

Vf = g*t = 9.8 m/s^2 t

Vf^2 = 2g(altura inicial) = 2*9.8m/s^2 * 705.5 m

Vf = 117.6 m/s

respuesta: => t = [117.6m/s] / [9.8m/s^2] = 12 s


Por favor ayúdenme, ya nose como hacer en Python esto.

Una inquietud: El enunciado no te da como dato la velocidad del lanzamiento del cohete, por tanto no puedes asumir que la velocidad inicial (velocidad de despegue) es cero. Has formulado con velocidad de lanzamiento igual a cero...verdad???...parece faltara como dato, la velocidad de despegue del cohete (velocidad inicial).