Como criar facilmente mapas de calor em python

Suponha que temos o seguinte conjunto de dados em Python que exibe o número de vendas feitas por uma determinada loja em cada dia da semana durante cinco semanas:

 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

Crie um mapa de calor básico

Podemos criar um mapa de calor básico usando a função ns.heatmap() :

 sns.heatmap(df) 

A barra de cores do lado direito exibe uma legenda indicando os valores representados pelas diferentes cores.

Adicione linhas ao mapa de calor

Você pode adicionar linhas entre quadrados no mapa de calor usando o argumento linewidths :

 sns.heatmap(df, linewidths=.5) 

Adicione anotações ao mapa de calor

Você também pode adicionar anotações ao mapa de calor usando o argumento annot=True :

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

Ocultar barra de cores do mapa de calor

Você também pode ocultar totalmente a barra de cores usando a opção cbar=False :

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

Alterar tema de cor do mapa de calor

Você também pode alterar o tema de cores usando o argumento cmap . Por exemplo, você pode definir as cores para variar do amarelo ao verde e ao azul:

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

Ou você pode mudar as cores do vermelho para o azul:

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

Para obter uma lista completa de paletas de cores, consulte a documentação do matplotlib .