Как создать тепловые карты с помощью seaborn (с примерами)

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

В этом руководстве объясняется, как создавать тепловые карты с использованием библиотеки визуализации Seaborn Python со следующим набором данных:

 #import seaborn
import seaborn as sns

#load "flights" dataset
data = sns. load_dataset (“ flights ”)
data = data. pivot (" month ", " year ", " passengers ")

#view first five rows of dataset
data. head ()

Создайте базовую тепловую карту

Мы можем использовать следующий синтаксис для создания базовой тепловой карты для этого набора данных:

 sns. heatmap (data)

Чай

Отрегулируйте размер тепловой карты

Мы можем использовать аргумент figsize , чтобы настроить общий размер тепловой карты:

 #set heatmap size
import matplotlib. pyplot as plt
plt. figure (figsize = (12.8))

#create heatmap
sns. heatmap (data) 

Изменение цветов тепловой карты

Мы можем использовать аргумент cmap , чтобы изменить цвета, используемые в тепловой карте. Например, мы могли бы выбрать цветовую палитру «Спектральная»:

 sns. heatmap (data, cmap=" Spectral ")

Или мы могли бы выбрать «холодную теплую» цветовую карту:

 sns. heatmap (data, cmap=" coolwarm ") 

Полный список доступных параметров cmap можно найти здесь .

Аннотируйте тепловую карту

Мы можем использовать следующий синтаксис, чтобы аннотировать каждую ячейку тепловой карты целочисленным форматированием и указать размер шрифта:

 sns. heatmap (data, annot= True , fmt=" d ", annot_kws={" size ":13}) 

Изменение цветовой панели тепловой карты

Наконец, мы можем отключить цветовую панель, если хотим использовать аргумент cbar :

 sns. heatmap (data, cbar= False ) 

Дополнительные обучающие материалы по Seaborn вы найдете на этой странице .