Hoe u de beste pasvorm in python tekent (met voorbeelden)
U kunt de volgende basissyntaxis gebruiken om een best passende lijn in Python te tekenen:
#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)
Het volgende voorbeeld laat zien hoe u deze syntaxis in de praktijk kunt gebruiken.
Voorbeeld 1: De best passende basislijn in Python uitzetten
De volgende code laat zien hoe u een best passende basislijn tekent in 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)
Voorbeeld 2: De best passende aangepaste lijn tekenen in Python
De volgende code laat zien hoe u dezelfde best passende lijn kunt maken als in het vorige voorbeeld, met de volgende toevoegingen:
- Aangepaste kleuren voor punten en best passende lijn
- Aangepaste stijl en breedte voor de best passende lijn
- De vergelijking van de aangepaste regressielijn die op de grafiek wordt weergegeven
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 )
Voel je vrij om de gepaste regressievergelijking in de gewenste (x,y) -coördinaten op de plot te plaatsen.
Voor dit specifieke voorbeeld hebben we gekozen voor (x, y) = (1, 17).
Aanvullende bronnen
In de volgende tutorials wordt uitgelegd hoe u verschillende regressiemodellen in Python kunt passen:
Een complete gids voor lineaire regressie in Python
Hoe polynomiale regressie uit te voeren in Python
Hoe kwantielregressie uit te voeren in Python