Matplotlib Ellipsen-Sammlung

Einführung

Matplotlib ist eine leistungsstarke Datenvisualisierungsbibliothek in Python. In diesem Lab werden wir die Verwendung von EllipseCollection zur Zeichnung einer Sammlung von Ellipsen untersuchen.

Tipps für die VM

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 nutzen.

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.

import matplotlib.pyplot as plt
import numpy as np

from matplotlib.collections import EllipseCollection

Daten für Ellipsen erstellen

Wir erstellen die Daten für unsere Ellipsen in Form von Arrays von x-Koordinaten, y-Koordinaten, Breite, Höhe und Winkel.

x = np.arange(10)
y = np.arange(15)
X, Y = np.meshgrid(x, y)

XY = np.column_stack((X.ravel(), Y.ravel()))

ww = X / 10.0
hh = Y / 15.0
aa = X * 9

Ellipsen-Sammlung erstellen

Wir erstellen eine EllipseCollection mit den oben genannten Daten und legen die Maßeinheiten auf 'x' und die Verschiebungen auf XY fest.

fig, ax = plt.subplots()

ec = EllipseCollection(ww, hh, aa, units='x', offsets=XY,
                       offset_transform=ax.transData)

Die Farbe der Ellipsen festlegen

Wir legen die Farbe jeder Ellipse in der EllipseCollection basierend auf der Summe ihrer x- und y-Koordinaten fest.

ec.set_array((X + Y).ravel())

Sammlung zur Grafik hinzufügen

Wir fügen die EllipseCollection zur Grafik hinzu.

ax.add_collection(ec)
ax.autoscale_view()
ax.set_xlabel('X')
ax.set_ylabel('y')
cbar = plt.colorbar(ec)
cbar.set_label('X+Y')
plt.show()

Zusammenfassung

In diesem Lab haben wir gelernt, wie man EllipseCollection in Matplotlib verwendet, um eine Sammlung von Ellipsen zu zeichnen. Wir haben auch gelernt, wie man die Farbe jeder Ellipse basierend auf ihren x- und y-Koordinaten setzt.