Jak narysować linię najlepszego dopasowania w pythonie (z przykładami)

Aby narysować najlepiej dopasowaną linię w Pythonie, możesz użyć następującej podstawowej składni:

 #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)

Poniższy przykład pokazuje, jak zastosować tę składnię w praktyce.

Przykład 1: Wykreślanie linii bazowej najlepszego dopasowania w Pythonie

Poniższy kod pokazuje, jak narysować najlepiej dopasowany punkt odniesienia w Pythonie:

 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) 

Przykład 2: Rysowanie niestandardowej linii najlepszego dopasowania w Pythonie

Poniższy kod pokazuje, jak utworzyć tę samą najlepiej dopasowaną linię, co w poprzednim przykładzie, z następującymi dodatkami:

• Niestandardowe kolory punktów i linii najlepszego dopasowania
• Niestandardowy styl i szerokość dla najlepszego dopasowania
• Równanie dopasowanej linii regresji wyświetlone na wykresie

 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 ) 

Możesz umieścić dopasowane równanie regresji w dowolnych współrzędnych (x, y) na wykresie.

W tym konkretnym przykładzie wybieramy (x, y) = (1, 17).

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak dopasować różne modele regresji w Pythonie:

Kompletny przewodnik po regresji liniowej w Pythonie
Jak wykonać regresję wielomianową w Pythonie
Jak wykonać regresję kwantylową w Pythonie