Библиотека Seaborn

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

Как сделать тепловые карты с Seaborn (с примерами)
В: Библиотека 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)
Тепловая карта Seaborn

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

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

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

#set heatmap size
import matplotlib.pyplot as plt
plt.figure(figsize = (12,8))

#create heatmap
sns.heatmap(data) 
Тепловая карта Seaborn настроить размер

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

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

sns.heatmap(data, cmap=" Spectral ")
Тепловая карта Seaborn с аргументом cmap

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

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

Полный список опций cmap доступен здесь .

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

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

sns.heatmap(data, annot= True , fmt=" d", annot_kws={" size ":13}) 
Тепловая карта Seaborn с аннотациями

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

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

sns.heatmap(data, cbar= False ) 
Тепловая карта Seaborn без цветовой полосы

Найдите больше руководств Seaborn на этой странице .

Еще от кодкамп
Замечательно! Вы успешно подписались.
Добро пожаловать обратно! Вы успешно вошли
Вы успешно подписались на кодкамп.
Срок действия вашей ссылки истек.
Ура! Проверьте свою электронную почту на наличие волшебной ссылки для входа.
Успех! Ваша платежная информация обновлена.
Ваша платежная информация не была обновлена.