Як створити теплові карти за допомогою 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 , щоб змінити кольори, які використовуються на тепловій карті. Наприклад, ми можемо вибрати палітру кольорів «Spectral»:

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

Або ми можемо вибрати «холодну» кольорову картку:

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

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

Додайте примітки до карти тепла

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

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

Змініть кольорову панель Heatmap

Нарешті, ми можемо вимкнути кольорову панель, якщо хочемо використовувати аргумент cbar :

 sns. heatmap (data, cbar= False ) 

Знайдіть більше підручників Seaborn на цій сторінці .