# -*- coding: utf-8 -*-
"""
Created on Thu Mar 21 09:18:51 2024

@author: pjaub
"""

import numpy as np
import matplotlib.pyplot as plt

plt.close('all')

# Données d'exemple
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.5, 3.3, 3.4, 5.5, 6.1])
errors = np.array([0.2, 0.3, 0.4, 0.3, 0.2]) 


coefficients = np.polyfit(x, y, deg=1)
y_d = np.polyval(coefficients, x)

# Affichage des résultats

plt.plot(x, y_d, color='k', label='Régression linéaire')
plt.plot(x, y, 'k+')
plt.errorbar(x, y, yerr=errors, color = 'r', 
             fmt = 'none', capsize = 3, elinewidth = 1, capthick = 1, label = 'écart-type')

plt.xlabel('masse ajoutée en grammes')
plt.ylabel('hauteurs en cm')
plt.title('Régression linéaire avec barres d\'erreur')
plt.legend()
plt.grid()
plt.show()