Как добавить заголовок к тепловой карте 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

Предположим, у нас есть следующий DataFrame 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 : Расположение текста заголовка.
• color : Цвет текста заголовка.
• 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