So erstellen sie eine manuelle legende in matplotlib (mit beispiel)

Sie können die Funktionen der Untermodule matplotlib.lines und matplotlib.patches verwenden, um eine manuelle Legende in einem Matplotlib-Plot zu erstellen.

Das folgende Beispiel zeigt, wie das geht.

Beispiel: Erstellen Sie eine manuelle Legende in Matplotlib

Der folgende Code zeigt, wie man in Matplotlib ein Streudiagramm mit einer Standardlegende erstellt:

 import matplotlib. pyplot as plt

#define data to plot
x = [1, 2, 3, 4, 5, 6, 7]
y = [2, 3, 5, 8, 12, 18, 27]

#create scatter plot of x vs. y
plt. scatter (x, y, label=' Original Data ', color=' steelblue ')

#add legend
plt. legend ()

#displayplot
plt. show () 

Um eine manuelle Legende mit benutzerdefinierten Linien und Quadraten zu erstellen, müssen wir die Untermodule matplotlib.lines und matplotlib.patches importieren.

Der folgende Code zeigt, wie diese Submodule zum Erstellen einer manuellen Legende verwendet werden:

 import matplotlib. pyplot as plt
from matplotlib. lines import Line2D
import matplotlib. patches as mpatches

#define data to plot
x = [1, 2, 3, 4, 5, 6, 7]
y = [2, 3, 5, 8, 12, 18, 27]

#create scatter plot of x vs. y
plt. scatter (x, y, label=' Original Data ', color=' steelblue ')

#define handles and labels that will get added to legend
handles, labels = plt. gca (). get_legend_handles_labels ()

#define patches and lines to add to legend
patch1 = mpatches. Patch (color=' orange ', label=' First Manual Patch ')
patch2 = mpatches. Patch (color=' orange ', label=' First Manual Patch ')   
line1 = Line2D([0], [0], label=' First Manual Line ', color=' purple ')
line2 = Line2D([0], [0], label=' Second Manual Line ', color=' red ')

#add handles
handles. extend ([patch1, line1, line2])

#add legend
plt. legend (handles=handles)

#displayplot
plt. show () 

Beachten Sie, dass diese Legende die Beschriftung der Originaldaten, aber auch die Beschriftungen und Formen der Elemente enthält, die wir manuell hinzugefügt haben.

Um die Beschriftungen oder Farben eines der Elemente zu ändern, ändern Sie einfach die Werte der Beschriftungs- und Farbargumente im vorherigen Codeabschnitt.

Hinweis : In diesem Tutorial erfahren Sie, wie Sie die Position der Legende im Diagramm ändern.

Zusätzliche Ressourcen

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

So erhöhen Sie die Plotgröße in Matplotlib
So passen Sie die Titelposition in Matplotlib an
So legen Sie Achsenbereiche in Matplotlib fest