So fügen sie einer darstellung in matplotlib eine durchschnittslinie hinzu

Sie können die folgende grundlegende Syntax verwenden, um einer Darstellung in Matplotlib eine Durchschnittslinie hinzuzufügen:

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

Beachten Sie, dass axhline dem Diagramm eine horizontale Linie hinzufügt und nanmean den Durchschnittswert berechnet (ohne Berücksichtigung von NaNs), an der die Linie platziert werden soll.

Das folgende Beispiel zeigt, wie diese Syntax in der Praxis verwendet wird.

Beispiel: Fügen Sie dem Plot in Matplotlib eine Durchschnittslinie hinzu

Angenommen, wir haben den folgenden Pandas-DataFrame:

 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

Mit dem folgenden Code können wir ein Streudiagramm von x gegenüber y erstellen und eine horizontale Linie hinzufügen, die den durchschnittlichen y-Wert darstellt:

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

Wir können sehen, dass dem Diagramm knapp über dem y-Wert von 8 eine Durchschnittslinie hinzugefügt wurde.

Wenn wir den durchschnittlichen y-Wert berechnen, werden wir feststellen, dass er 8,333 beträgt:

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

8.333333333

Beachten Sie, dass wir auch die Argumente color , linestyle und linewidth verwenden können, um die Farbe, den Linientyp bzw. die Linienbreite der Durchschnittslinie anzugeben:

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

Hinweis : Die vollständige Online-Dokumentation für die Funktion axhline() finden Sie hier .

Zusätzliche Ressourcen

Die folgenden Tutorials erklären, wie Sie andere häufige Aufgaben in Matplotlib ausführen:

So fügen Sie eine Trendlinie in Matplotlib hinzu
So zeichnen Sie eine vertikale Linie in Matplotlib
So zeichnen Sie mehrere Linien in Matplotlib