Як додати назву до теплової карти seaborn (з прикладом)

Ви можете використовувати наступний базовий синтаксис, щоб додати заголовок до карти тепла в Seaborn:

 import matplotlib. pyplot as plt
import seaborn as sns

#create heatmap
sns. heatmap (df)

#add title
plt. title (' This is my title ')

У наступному прикладі показано, як використовувати цей синтаксис на практиці.

Приклад: додайте назву до карти тепла в Seaborn

Припустімо, що ми маємо такий фрейм даних pandas, який містить інформацію про очки, набрані різними баскетболістами за п’ять років поспіль:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' year ': [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5],
                   ' player ': ['A', 'A', 'A', 'A', 'A', 'B', 'B',
                              'B', 'B', 'B', 'C', 'C', 'C', 'C', 'C'],
                   ' points ': [8, 12, 14, 14, 15, 10, 15, 19, 29, 13,
                              10, 14, 22, 24, 25]})

#pivot DataFrame
df = df. pivot (' player ', ' year ', ' points ')

#view DataFrame
print (df)

year 1 2 3 4 5
player                    
A 8 12 14 14 15
B 10 15 19 29 13
C 10 14 22 24 25

Якщо ми використовуємо функцію heatmap() для створення карти тепла в seaborn, за замовчуванням до карти тепла не буде додано заголовок:

 import seaborn as sns

#create heatmap
sns. heatmap (df, linewidth= .3 )

Однак ми можемо використати функцію title() matplotlib, щоб швидко додати заголовок до теплової карти:

 import matplotlib. pyplot as plt
import seaborn as sns

#create heatmap
sns. heatmap (df, linewidth= .3 )

#add title to heatmap
plt. title (' Points Scored by Players Each Year ')

Також зауважте, що ми можемо використовувати такі аргументи у функції title(), щоб змінити вигляд заголовка:

• loc : розташування тексту заголовка
• колір : колір тексту заголовка
• size : розмір шрифту тексту заголовка

Наступний код показує, як додати заголовок, вирівняний за лівим краєм, із червоним кольором шрифту та розміром 14:

 import matplotlib. pyplot as plt
import seaborn as sns

#create heatmap
sns. heatmap (df, linewidth= .3 )

#add customized title to heatmap
plt. title (' Points Scored by Players Each Year ', loc=' left ', color=' red ', size= 14 ) 

Не соромтеся змінювати аргументи функції title() , щоб створити саме той заголовок, який ви хочете.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в Seaborn:

Як налаштувати розмір теплових карт у Seaborn
Як додати право власності на ділянки Seaborn
Як створити підсюжети в Seaborn