Hoe u een gemiddelde lijn kunt toevoegen aan een plot in matplotlib

U kunt de volgende basissyntaxis gebruiken om een gemiddelde lijn toe te voegen aan een plot in Matplotlib:

 import matplotlib. pyplot as plt
import numpy as np

#create scatterplot
plt. scatter (df. x , df. y )

#add horizontal line at mean value of y
plt. axhline (y=np. nanmean (df. y ))

Merk op dat axhline een horizontale lijn aan de grafiek toevoegt en dat nanmean de gemiddelde waarde berekent (waarbij NaN’s worden genegeerd) waar de lijn moet worden geplaatst.

Het volgende voorbeeld laat zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld: Voeg een gemiddelde lijn toe aan de plot in Matplotlib

Stel dat we de volgende panda’s DataFrame hebben:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' x ': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
                   ' y ':[2, 5, 6, 5, 7, 8, 10, 12, 10, 9, 11, 15]})

#view first five rows of DataFrame
df. head ()

	x y
0 1 2
1 2 5
2 3 6
3 4 5
4 5 7

We kunnen de volgende code gebruiken om een spreidingsdiagram van x versus y te maken en een horizontale lijn toe te voegen die de gemiddelde y-waarde vertegenwoordigt:

 import matplotlib. pyplot as plt
import numpy as np

#create scatterplot
plt. scatter (df. x , df. y )

#add horizontal line at mean value of y
plt. axhline (y=np. nanmean (df. y )) 

We kunnen zien dat er een gemiddelde lijn aan de grafiek is toegevoegd net boven de y-waarde van 8.

Als we de gemiddelde y-waarde berekenen, zullen we ontdekken dat deze 8,333 is:

 #calculate average y-value
n.p. nanmean (df. y )

8.333333333

Merk op dat we ook de argumenten color , linestyle en linewidth kunnen gebruiken om respectievelijk de kleur, het lijntype en de lijnbreedte van de gemiddelde lijn op te geven:

 import matplotlib. pyplot as plt
import numpy as np

#create scatterplot
plt. scatter (df. x , df. y )

#add horizontal line at mean value of y
plt. axhline (y=np. nanmean (df. y ), color=' red ', linestyle=' -- ', linewidth= 3 , label=' Avg ') 

Opmerking : u kunt hier de volledige online documentatie voor de functie axhline() vinden.

Aanvullende bronnen

In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende taken in Matplotlib kunt uitvoeren:

Hoe een trendlijn toe te voegen in Matplotlib
Hoe teken je een verticale lijn in Matplotlib?
Hoe meerdere lijnen in Matplotlib te plotten