import numpy as np
import matplotlib.pyplot as plt

#Liste des demi-grand axe en unité astronomique
a_ua=[0.387,0.723,1,1.524,5.203,9.555,19.22,30.10]
#Liste des demi-grand axe en m
a=[a_ua[i]*149.6e9 for i in range(len(a_ua))]
#Liste des périodes en jours
T_jours=[87.97,224.7,365.26,365.26+322,11*365.26+315,29*365.26+167,84*365.26+27,164*365.26+280]
#Liste des périodes en secondes
T=[T_jours[i]*24*3600 for i in range(len(T_jours))]

#Liste des T^2
T2=[T[i]**2 for i in range(len(T))]

#Liste des a^3
a3=[a[i]**3 for i in range(len(a))]

# Courbe a^3 en fonction de T^2
plt.plot(T2,a3)
plt.xlabel("T^2 (s^2)")
plt.ylabel("a^3 (m^3)")
plt.grid()
plt.show()
# Régression linéaire
pente,ordao = np.polyfit(T2,a3,1)
# Obtention de la masse du Soleil
print("masse Soleil=",pente*4*(np.pi)**2/6.67e-11,"kg")