Come creare facilmente mappe di calore in python

Supponiamo di avere il seguente set di dati in Python che mostra il numero di vendite effettuate da un determinato negozio ogni giorno della settimana per cinque settimane:

 import numpy as np
import pandas as pd 
import seaborn as sns

#create a dataset
np.random.seed(0)
data = {'day': np.tile(['Mon', 'Tue', 'Wed', 'Thur', 'Fri'], 5),
        'week': np.repeat([1, 2, 3, 4, 5], 5),
        'sales': np.random.randint(0, 50, size=25)
        }

df = pd.DataFrame(data,columns=['day','week','sales'])
df = df.pivot('day', 'week', 'sales')

view first ten rows of dataset
df[:10]

week 1 2 3 4 5
day					
Fri 3 36 12 46 13
My 44 39 23 1 24
Thursday 3 21 24 23 25
Kills 47 9 6 38 17
Wed 0 19 24 39 37

Crea una mappa termica di base

Possiamo creare una mappa termica di base utilizzando la funzione ns.heatmap() :

 sns.heatmap(df) 

La barra dei colori sul lato destro mostra una legenda che indica i valori rappresentati dai diversi colori.

Aggiungi linee alla mappa termica

Puoi aggiungere linee tra i quadrati nella mappa termica utilizzando l’argomento linewidths :

 sns.heatmap(df, linewidths=.5) 

Aggiungi annotazioni alla mappa termica

Puoi anche aggiungere annotazioni alla mappa termica utilizzando l’argomento annot=True :

 sns.heatmap(df, linewidths=.5, annot=True) 

Nascondi la barra dei colori della mappa termica

Puoi anche nascondere completamente la barra dei colori utilizzando l’opzione cbar=False :

 sns.heatmap(df, linewidths=.5, annot=True, cbar=False) 

Cambia il tema del colore della mappa termica

Puoi anche cambiare il tema del colore usando l’argomento cmap . Ad esempio, puoi impostare i colori in modo che vadano dal giallo al verde al blu:

 sns.heatmap(df, cmap='YlGnBu')

Oppure potresti avere i colori che vanno dal rosso al blu:

 sns.heatmap(df, cmap='RdBu') 

Per un elenco completo delle tavolozze dei colori, consultare la documentazione di matplotlib .