Тепловые карты корреляций

Матрица из двадцати признаков — это сплошная стена чисел. Тепловая карта раскрашивает каждую ячейку, и тесные связи сразу бросаются в глаза. Восемь дней в кинотеатре:

Этот код отображает её:

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 задают границы цветовой шкалы: красный — это 1, белый — 0, синий — -1. fig.colorbar(im) добавляет шкалу, а set_xticks(positions, labels) задаёт названия ячеек.

Билеты и попкорн вместе светятся красным; тёплые дни отмечены синим на фоне обоих. Количество сотрудников почти не имеет значения.