Comment ajouter un titre à la carte thermique Seaborn (avec exemple)

Vous pouvez utiliser la syntaxe de base suivante pour ajouter un titre à une carte thermique dans Seaborn :

import matplotlib.pyplot as plt
import seaborn as sns

#create heatmap
sns.heatmap(df)

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

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : ajouter un titre à la carte thermique dans Seaborn

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur les points marqués par divers joueurs de basket-ball au cours de cinq années consécutives :

import pandas as pd

#create DataFrame
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

Si nous utilisons la fonction heatmap() pour créer une heatmap dans seaborn, aucun titre ne sera ajouté à la heatmap par défaut :

import seaborn as sns

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

Cependant, nous pouvons utiliser la fonction title() de matplotlib pour ajouter rapidement un titre à la heatmap :

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

Notez également que nous pouvons utiliser les arguments suivants au sein de la fonction title() pour modifier l’apparence du titre :

• loc : Emplacement du texte du titre
• color : Couleur du texte du titre
• size : Taille de la police du texte du titre

Le code suivant montre comment ajouter un titre aligné à gauche, avec une couleur de police rouge et une taille de police de 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)

N’hésitez pas à modifier les arguments de la fonction title() pour créer le titre exact que vous souhaitez.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans Seaborn :

Comment ajuster la taille des cartes thermiques dans Seaborn
Comment ajouter un titre aux parcelles Seaborn
Comment créer des sous-parcelles dans Seaborn