import numpy as np
import matplotlib.pyplot as plt

# Création des tableaux avec les valeurs expérimentales
P_A0 = np.array([])
v_0 = np.array([])


X = 
Y = 

# Régression linéaire Y = a*X + B

p = np.polyfit(X, Y, 1)
pente = p[0]             # pente de la droite
ord_origine = p[1]       # ordonnée à l’origine
Rcarre = np.corrcoef(X, Y)[0, 1]**2  # coefficient de corrélation au carré

# Tracé des points expérimentaux et de la droite de régression
plt.plot(X, np.polyval(p, X), "--", label="Régression linéaire")
plt.plot(X, Y , "+", label="Données expérimentales")

# Légendes
plt.xlabel("", size=14)
plt.ylabel("", size=14)

# Affichage des valeurs numériques
plt.text(9.1, 6.0, f"Pente : {pente:.2f}", fontsize=10, color="black")
plt.text(9.1, 5.75, f"Ordonnée à l'origine : {ord_origine:.2f}", fontsize=10, color="black")
plt.text(9.1, 5.5, f"$R^2$ : {Rcarre:.4f}", fontsize=10, color="black")

plt.legend()
plt.show()
plt.grid()