print("Fisica computacional (Clase 2:00-3:00)")
print("Alumna: Nalleli Iridian Avila Garcia 1941603")
print("Este programa calcula las velocidades de caída libre de un objeto con v0=0m/s, c=12.5kg/s, en un tiempo de 0 a 64 s, con dt=2 seg.")

import numpy as np
import matplotlib.pyplot as plt

# Parámetros
m = 68.1  # masa del objeto en kg
g = 9.81  # aceleración debido a la gravedad en m/s^2
c = 12.5  # coeficiente de arrastre
t_final = 64  # tiempo final en segundos
dt = 2  # tamaño del paso de tiempo en segundos

# Inicialización de las variables
t = np.arange(0, t_final + dt, dt)  # vector de tiempo
v = np.zeros_like(t)  # vector de velocidad

# Método de Euler
for i in range(1, len(t)):
    dv = (g - c/m * v[i-1]) * dt
    v[i] = v[i-1] + dv

# Imprimir cada velocidad
for i in range(len(t)):
    print(f"En el tiempo t={t[i]:.2f} s, la velocidad es {v[i]:.2f} m/s.")

# Gráfica de los resultados
plt.figure(figsize=(10, 6))
plt.plot(t, v, label='Velocidad (m/s)')
plt.xlabel('Tiempo (s)')
plt.ylabel('Velocidad (m/s)')
plt.title('Velocidad de un objeto en caída libre')
plt.legend()
plt.grid(True)
plt.show()