Erstellen einer Legende mit vordefinierten Etiketten

Einführung

In der Datenvisualisierung ist eine Legende der Schlüssel zur Interpretation der visuellen Elemente eines Diagramms. Sie hilft dem Betrachter, die Daten und die Bedeutung der visuellen Darstellung zu verstehen. Matplotlib ist eine beliebte Python-Bibliothek zur Erstellung von Datenvisualisierungen, einschließlich Diagrammen mit Legenden. In diesem Tutorial lernen wir, wie man in Matplotlib eine Legende mit vordefinierten Beschriftungen erstellt.

VM-Tipps

Nachdem der VM-Start abgeschlossen ist, klicken Sie in der oberen linken Ecke, um zur Registerkarte Notebook zu wechseln und Jupyter Notebook für die Übung zu öffnen.

Manchmal müssen Sie einige Sekunden warten, bis Jupyter Notebook vollständig geladen ist. Die Validierung von Vorgängen kann aufgrund von Einschränkungen in Jupyter Notebook nicht automatisiert werden.

Wenn Sie bei der Lernphase Probleme haben, können Sie Labby gerne fragen. Geben Sie nach der Sitzung Feedback, und wir werden das Problem für Sie prompt beheben.

Importieren der erforderlichen Bibliotheken

Wir beginnen mit dem Importieren der erforderlichen Bibliotheken, die Matplotlib und NumPy einschließen. Wir verwenden NumPy, um einige simulierte Daten für unser Diagramm zu generieren.

import matplotlib.pyplot as plt
import numpy as np

Generieren der Daten

Als nächstes werden wir einige simulierte Daten generieren, die wir in unserem Diagramm verwenden. Wir werden zwei Arrays, a und b, mit der NumPy-Funktion arange erstellen. Anschließend berechnen wir zwei weitere Arrays, c und d, wobei wir die exp-Funktion verwenden, um die Exponentialwerte von a zu berechnen und d als das Umgekehrte von c zu berechnen.

## Make some fake data.
a = b = np.arange(0, 3,.02)
c = np.exp(a)
d = c[::-1]

Erstellen des Diagramms

Jetzt sind wir bereit, unser Diagramm zu erstellen. Wir werden die plot-Funktion von Matplotlib verwenden, um drei Linien auf dem gleichen Graphen zu zeichnen, wobei jeder eine vordefinierte Bezeichnung hat. Wir werden das label-Parameter verwenden, um den Bezeichnungen jeder Linie zuzuweisen.

## Create plots with pre-defined labels.
fig, ax = plt.subplots()
ax.plot(a, c, 'k--', label='Model length')
ax.plot(a, d, 'k:', label='Data length')
ax.plot(a, c + d, 'k', label='Total message length')

Hinzufügen der Legende

Um der Legende zu unserem Diagramm hinzuzufügen, verwenden wir die legend-Funktion von Matplotlib. Wir übergeben das loc-Parameter, um den Standort der Legende anzugeben, und den shadow-Parameter, um einem Schatteneffekt zur Legende hinzuzufügen. Wir verwenden auch den fontsize-Parameter, um die Schriftgröße der Legende festzulegen.

legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')

Gestaltung der Legende

Schließlich können wir die Legende gestalten, um sie visuell ansprechender zu machen. Wir verwenden die get_frame-Funktion, um den Rahmen der Legende zu erhalten, und anschließend die set_facecolor-Funktion, um die Hintergrundfarbe des Rahmens festzulegen.

## Put a nicer background color on the legend.
legend.get_frame().set_facecolor('C0')

Anzeigen des Diagramms

Wir können jetzt das Diagramm mit der show-Funktion von Matplotlib anzeigen.

plt.show()

Zusammenfassung

In diesem Tutorial haben wir gelernt, wie man in Matplotlib eine Legende mit vordefinierten Bezeichnungen erstellt. Wir haben die plot-Funktion verwendet, um drei Linien auf dem gleichen Graphen zu zeichnen und den label-Parameter verwendet, um jeder Linie eine Bezeichnung zuzuweisen. Anschließend haben wir die legend-Funktion verwendet, um der Legende zum Diagramm hinzuzufügen und die Legende gestaltet, um sie visuell ansprechender zu machen. Indem Sie diese Schritte befolgen, können Sie für Ihre eigenen Diagramme in Matplotlib Legenden erstellen.