Ejercicio: Un paracaidista en caida libre usando Python y Matplotlib para graficar

Hola, les comparto un ejercicio que desarrollé para una clase que doy de métodos numéricos,

La idea es comprobar diferentes masas y diferentes coeficientes de resistencias para el ejemplo usar 12.5, caída libre y podemos calcular un valor cuatruple sí abrimos un paracaídas.

Para hacer el ejercicio debes tener instalado matplotlib y numpy, recuerden tener instalado PIP

#!/usr/bin/python3.5
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np

print('\n')
print('Practica 3')
print('')
print('-*-*-*-*-*-*-*-*-*-*-*-*-*-')
print('|Problema del paracaidista|')
print('-*-*-*-*-*-*-*-*-*-*-*-*-*-')

print('')
print('Language Version : Python 3.XX')
print('by: rusbel bermudez rivera')

print('\n')
print('Ingresa tus datos a utilizar para el experimento')
print('')

G = 9.8
t_inicial = 0
v_inicial = 0
idx = 0m = float(input('Ingresa el valor de la masa en kg: '))
c = float(input('Ingresa el valor del coeficiente de resistencia en kg/s: '))

delta_time = float(input('Ingresa el valor del delta a utilizar para los incrementos: '))
t_final = delta_time

last_point = int(input('¿Cuantos segmentos delta quieres analizar?:')) * delta_time

#ejes para graficar
x_axys = [t_inicial]
y_axys = [v_inicial]

#Uso del ciclo whilewhile t_final < last_point:
    v_final = v_inicial
    print('v_{} = {} m/s'.format(idx,round(v_final,4)))
    v_final = v_inicial + (G - ((c/m)*v_inicial)) * (t_final - t_inicial)

    x_axys.append(int(t_final))
    y_axys.append(v_final)
    v_inicial = v_final
    t_inicial = t_final
    t_final += delta_time
    idx += 1print(x_axys)
print(y_axys)
plt.plot(x_axys,y_axys)
plt.axis([0, 18, 0, 60])
plt.show()