Comment tracer la ligne la mieux adaptée en Python (avec exemples)

Vous pouvez utiliser la syntaxe de base suivante pour tracer une ligne la mieux ajustée en Python :

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y)

#add line of best fit to plot
plt.plot(x, a*x+b)

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple 1 : tracer la ligne de base du meilleur ajustement en Python

Le code suivant montre comment tracer une ligne de base la mieux adaptée en Python :

import numpy as np
import matplotlib.pyplot as plt

#define data
x = np.array([1, 2, 3, 4, 5, 6, 7, 8])
y = np.array([2, 5, 6, 7, 9, 12, 16, 19])

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y)

#add line of best fit to plot
plt.plot(x, a*x+b)        

Exemple 2 : tracer la ligne personnalisée la mieux adaptée en Python

Le code suivant montre comment créer la même ligne la mieux ajustée que l’exemple précédent, avec les ajouts suivants :

• Couleurs personnalisées pour les points et la ligne la mieux ajustée
• Style et largeur personnalisés pour la ligne la mieux ajustée
• L’équation de la droite de régression ajustée affichée sur le tracé

import numpy as np
import matplotlib.pyplot as plt

#define data
x = np.array([1, 2, 3, 4, 5, 6, 7, 8])
y = np.array([2, 5, 6, 7, 9, 12, 16, 19])

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y, color='purple')

#add line of best fit to plot
plt.plot(x, a*x+b, color='steelblue', linestyle='--', linewidth=2)

#add fitted regression equation to plot
plt.text(1, 17, 'y = ' + '{:.2f}'.format(b) + ' + {:.2f}'.format(a) + 'x', size=14)

N’hésitez pas à placer l’équation de régression ajustée dans les coordonnées (x, y) que vous souhaitez sur le tracé.

Pour cet exemple particulier, nous avons choisi (x, y) = (1, 17).

Ressources additionnelles

Les didacticiels suivants expliquent comment adapter différents modèles de régression en Python :

Un guide complet de la régression linéaire en Python
Comment effectuer une régression polynomiale en Python
Comment effectuer une régression quantile en Python