from scipy.integrate import odeint
import matplotlib.pyplot as plt
import numpy as np

# Constantes du système
g = 9.81  # m/s2
alpha = 1e-1  # kg/m
m = 1       # kg


# Définition de l'équation différentielle
def equation(v,t):  # Attention à l'inversion de t et v
    return g-(alpha/m)*v
t0 = 0
tf = 100  # seconde
v0 = 0  # Condition initiale
t = np.linspace(t0,tf,500) # Création des instants de calculs

# Résolution
v = odeint(equation,v0,t)
print(t)  # Affichage des instants t
print(v[:,0])  # Affichage des résultats

# Tracé de la vitesse au cours du temps
plt.plot(t,v[:,0])
plt.ylabel('v(t) en m/s')
plt.xlabel("Temps (s)")
plt.title("Vitesse lors d'une chute libre soumise à une force de frottement en v")
plt.grid(which='both')
plt.show()