Textpfeil annotieren

Einführung

In diesem Tutorial wird Ihnen die Erstellung eines annotierten Streudiagramms mit einem Textpfeil mithilfe von Matplotlib in Python gezeigt.

Tipps für die virtuelle Maschine

Nachdem der Start der virtuellen Maschine 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 der Einschränkungen von 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.

Bibliotheken importieren und zufällige Daten generieren

Zunächst müssen wir die erforderlichen Bibliotheken importieren und einige zufällige Daten für unser Streudiagramm generieren.

import matplotlib.pyplot as plt
import numpy as np

## Fixing random state for reproducibility
np.random.seed(19680801)

fig, ax = plt.subplots(figsize=(5, 5))
ax.set_aspect(1)

x1 = -1 + np.random.randn(100)
y1 = -1 + np.random.randn(100)
x2 = 1. + np.random.randn(100)
y2 = 1. + np.random.randn(100)

ax.scatter(x1, y1, color="r")
ax.scatter(x2, y2, color="g")

Textanmerkungen zum Diagramm hinzufügen

Als nächstes fügen wir Textanmerkungen zum Diagramm hinzu, indem wir die Funktion ax.text() verwenden. Wir erstellen zwei Anmerkungen, eine für "Sample A" und eine für "Sample B".

bbox_props = dict(boxstyle="round", fc="w", ec="0.5", alpha=0.9)
ax.text(-2, -2, "Sample A", ha="center", va="center", size=20,
        bbox=bbox_props)
ax.text(2, 2, "Sample B", ha="center", va="center", size=20,
        bbox=bbox_props)

Ein Textpfeil hinzufügen, um die Richtung anzuzeigen

Um die Richtung der Daten anzuzeigen, fügen wir einen Textpfeil hinzu, indem wir die Funktion ax.text() und den Parameter bbox mit boxstyle auf "rarrow" setzen.

bbox_props = dict(boxstyle="rarrow", fc=(0.8, 0.9, 0.9), ec="b", lw=2)
t = ax.text(0, 0, "Direction", ha="center", va="center", rotation=45,
            size=15,
            bbox=bbox_props)

bb = t.get_bbox_patch()
bb.set_boxstyle("rarrow", pad=0.6)

Diagrammgrenzen festlegen und das Diagramm anzeigen

Schließlich legen wir die x- und y-Grenzen des Diagramms fest und zeigen das Diagramm mit den Funktionen ax.set_xlim(), ax.set_ylim() und plt.show() an.

ax.set_xlim(-4, 4)
ax.set_ylim(-4, 4)

plt.show()

Zusammenfassung

In diesem Tutorial haben wir gelernt, wie man mit Matplotlib in Python ein annotiertes Streudiagramm mit einem Textpfeil erstellt. Wir haben die Funktion ax.text() verwendet, um Anmerkungen und einen Textpfeil zum Diagramm hinzuzufügen, und die Funktionen ax.set_xlim(), ax.set_ylim() und plt.show() verwendet, um die Diagrammgrenzen festzulegen und das Diagramm anzuzeigen.