import matplotlib.pyplot as plt
import numpy as np

# Masse (kg) :
m =

# Valeur de la vitesse d'entrée (m/s) :
v_0 =

# Poids (N) :
P =

# Poussée d'Archimède (N) :
P_A =

# Calcul de l'accélération
a = (P_A - P)/m

#Détermination des limites du graphe
t_max = 2*(v_0)/a
z_min = -v_0**2/(2*a)
# Valeurs de temps (abscisse)
t = np.linspace(0, t_max, 1000)
# Calcul de l'ordonnée
z = (1/2)*a*t**2 - (v_0)*t

# Dessin de la courbe
n_ticks = 10 # Modifie le nombre de points sur l'axe des abscisses

fig, ax = plt.subplots()
ax.plot(t, z)
plt.title("Profondeur du plongeur en fonction du temps")
ax.xaxis.set_ticks_position('top')
ax.xaxis.set_label_position('top')
ax.set_xticks(np.linspace(0, t_max, n_ticks))
ax.set_yticks(np.linspace(z_min-100, 0, n_ticks))
ax.set_xlabel("t(s)")
ax.set_ylabel("z(m)")
plt.show()