Peta Panas Korelasi

Matriks dengan dua puluh fitur adalah hamparan angka yang memenuhi layar. Heatmap memberi warna pada setiap sel, sehingga pasangan yang kuat terlihat jelas. Delapan hari di sebuah bioskop:

Kode ini menggambarnya:

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 menetapkan warna: merah adalah 1, putih 0, biru -1. fig.colorbar(im) menambahkan legenda, dan set_xticks(positions, labels) memberi nama pada sel.

Tiket dan popcorn sama-sama bersinar merah; hari yang hangat tampak biru jika dibandingkan dengan keduanya. Jumlah staf hampir tidak berpengaruh.