Как сделать тепловые карты с Seaborn (с примерами)

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

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

#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)

По оси X отображается год, по оси Y — месяц, а цвет квадратов на тепловой карте представляет количество полетов в этих конкретных комбинациях год-месяц.

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

Мы можем использовать аргумент 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 на этой странице .