Comment créer des cartes thermiques avec Seaborn (avec exemples)

Une carte thermique est un type de graphique qui utilise différentes nuances de couleurs pour représenter les valeurs des données.

Ce didacticiel explique comment créer des cartes thermiques à l’aide de la bibliothèque de visualisation Python Seaborn avec l’ensemble de données suivant :

#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()

Créer une carte thermique de base

Nous pouvons utiliser la syntaxe suivante pour créer une carte thermique de base pour cet ensemble de données :

sns.heatmap(data)

L’axe des X affiche l’année, l’axe des Y affiche le mois et la couleur des carrés dans la carte thermique représente le nombre de vols dans ces combinaisons année-mois particulières.

Ajuster la taille de la carte thermique

Nous pouvons utiliser l’argument figsize pour ajuster la taille globale de la carte thermique :

#set heatmap size
import matplotlib.pyplot as plt
plt.figure(figsize = (12,8))

#create heatmap
sns.heatmap(data)

Changer les couleurs de la carte thermique

Nous pouvons utiliser l’argument cmap pour modifier les couleurs utilisées dans la carte thermique. Par exemple, on pourrait choisir la palette de couleurs « Spectral » :

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

Ou nous pourrions choisir la carte de couleurs « coolwarm » :

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

Trouvez une liste complète des options cmap disponibles ici .

Annoter la Heatmap

Nous pouvons utiliser la syntaxe suivante pour annoter chaque cellule de la carte thermique avec un formatage entier et spécifier la taille de la police :

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

Modifier la barre de couleurs de la Heatmap

Enfin, nous pouvons désactiver la barre de couleurs si nous souhaitons utiliser l’argument cbar :

sns.heatmap(data, cbar=False)

Trouvez plus de didacticiels Seaborn sur cette page .