Hoe u eenvoudig hittekaarten kunt maken in python

Stel dat we de volgende dataset in Python hebben die vijf weken lang elke dag van de week het aantal verkopen van een bepaalde winkel weergeeft:

 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

Maak een eenvoudige hittekaart

We kunnen een eenvoudige heatmap maken met behulp van de ns.heatmap() functie:

 sns.heatmap(df) 

De kleurenbalk aan de rechterkant toont een legenda die de waarden aangeeft die door de verschillende kleuren worden weergegeven.

Voeg regels toe aan de heatmap

U kunt lijnen tussen vierkanten in de heatmap toevoegen met behulp van het lijnbreedte- argument:

 sns.heatmap(df, linewidths=.5) 

Voeg annotaties toe aan de heatmap

U kunt ook annotaties aan de heatmap toevoegen met behulp van het argument annot=True :

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

Verberg de kleurenbalk van de heatmap

Je kunt de kleurenbalk ook volledig verbergen met de optie cbar=False :

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

Wijzig het kleurthema van de heatmap

U kunt het kleurthema ook wijzigen met behulp van het cmap- argument. U kunt bijvoorbeeld instellen dat de kleuren variëren van geel via groen naar blauw:

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

Of je kunt de kleuren van rood naar blauw laten gaan:

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

Zie de matplotlib-documentatie voor een volledige lijst met kleurenpaletten.