Anpassbares Streudiagrammvisualisierung

Einführung

In diesem Tutorial lernen wir, wie man mit Python Matplotlib einen Streudiagramm mit variierenden Markerfarben und -größen erstellt.

Tipps für virtuelle Maschinen (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 ö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.

Importieren der erforderlichen Bibliotheken

Wir beginnen mit dem Importieren der erforderlichen Bibliotheken, nämlich Matplotlib und Numpy.

import matplotlib.pyplot as plt
import numpy as np

Daten laden

Wir werden ein numpy-Record-Array aus yahoo-csv-Daten laden, die Felder wie Datum, Öffnung, Hoch, Tief, Schluss, Volumen, adjustierter Schluss aus dem mpl-data/sample_data-Verzeichnis haben. Das Record-Array speichert das Datum als np.datetime64 mit einem Tagesformat ('D') in der Datumsspalte.

import matplotlib.cbook as cbook

price_data = cbook.get_sample_data('goog.npz')['price_data'].view(np.recarray)
price_data = price_data[-250:]  ## get the most recent 250 trading days

Werte für das Streudiagramm berechnen

Wir werden die Werte für Delta1, Volumen und Schluss für das Streudiagramm berechnen.

delta1 = np.diff(price_data.adj_close) / price_data.adj_close[:-1]

## Markergröße in Einheiten von Punkten^2
volume = (15 * price_data.volume[:-2] / price_data.volume[0])**2
close = 0.003 * price_data.close[:-2] / 0.003 * price_data.open[:-2]

Streudiagramm erstellen

Wir werden ein Streudiagramm mit variierenden Markerfarben und -größen unter Verwendung der berechneten Werte erstellen.

fig, ax = plt.subplots()
ax.scatter(delta1[:-1], delta1[1:], c=close, s=volume, alpha=0.5)

ax.set_xlabel(r'$\Delta_i$', fontsize=15)
ax.set_ylabel(r'$\Delta_{i+1}$', fontsize=15)
ax.set_title('Volume und prozentuale Änderung')

ax.grid(True)
fig.tight_layout()

plt.show()

Zusammenfassung

Wir haben gelernt, wie man mit Python Matplotlib ein Streudiagramm mit variierenden Markerfarben und -größen erstellt. Wir begannen mit dem Importieren der erforderlichen Bibliotheken, ladeten dann Daten, berechneten die Werte für das Streudiagramm und erstellten das Streudiagramm.