Como adicionar um título ao seaborn heatmap (com exemplo)

Você pode usar a seguinte sintaxe básica para adicionar um título a um mapa de calor no Seaborn:

 import matplotlib. pyplot as plt
import seaborn as sns

#create heatmap
sns. heatmap (df)

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

O exemplo a seguir mostra como usar essa sintaxe na prática.

Exemplo: Adicione um título ao mapa de calor no Seaborn

Suponha que temos o seguinte DataFrame do pandas que contém informações sobre os pontos marcados por vários jogadores de basquete em cinco anos consecutivos:

 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

Se usarmos a função heatmap() para criar um mapa de calor no mar, nenhum título será adicionado ao mapa de calor por padrão:

 import seaborn as sns

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

No entanto, podemos usar a função title() do matplotlib para adicionar rapidamente um título ao mapa de calor:

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

Observe também que podemos usar os seguintes argumentos na função title() para alterar a aparência do título:

• loc : Localização do texto do título
• color : Cor do texto do título
• size : Tamanho da fonte do texto do título

O código a seguir mostra como adicionar um título alinhado à esquerda, com uma cor de fonte vermelha e um tamanho de fonte 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 ) 

Sinta-se à vontade para modificar os argumentos da função title() para criar o título exato que você deseja.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no Seaborn:

Como ajustar o tamanho dos mapas de calor no Seaborn
Como adicionar um título aos gráficos Seaborn
Como criar subtramas em Seaborn