Korrelations-Heatmaps

Eine Matrix mit zwanzig Merkmalen ist eine Zahlenwand. Eine Heatmap färbt jede Zelle ein, sodass die starken Paare sofort auffallen. Die acht Tage eines Kinos:

Dieser Code zeichnet es:

cinema = pd.DataFrame({
    "tickets": [120, 340, 210, 410, 180, 290, 380, 150],
    "popcorn": [70, 150, 100, 200, 95, 120, 170, 60],
    "temp_c": [24, 12, 15, 9, 20, 19, 11, 22],
    "staff": [6, 5, 8, 7, 5, 6, 6, 8],
})
corr = cinema.corr()
fig, ax = plt.subplots()
im = ax.imshow(corr, cmap="coolwarm", vmin=-1, vmax=1)
fig.colorbar(im)
ax.set_xticks(range(4), corr.columns)
ax.set_yticks(range(4), corr.columns)
ax.set_title("Cinema Correlations")
plt.show()

vmin=-1, vmax=1 legt die Farben fest: Rot ist 1, Weiß 0, Blau -1. fig.colorbar(im) fügt die Legende hinzu, und set_xticks(positions, labels) benennt die Zellen.

Tickets und Popcorn leuchten gemeinsam rot; warme Tage sind im Vergleich zu beiden blau. Die Personalbesetzung spielt kaum eine Rolle.