Como criar mapas de calor com seaborn (com exemplos)


Um mapa de calor é um tipo de gráfico que usa diferentes tons de cores para representar valores de dados.

Este tutorial explica como criar mapas de calor usando a biblioteca de visualização Seaborn Python com o seguinte conjunto de dados:

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

Crie um mapa de calor básico

Podemos usar a seguinte sintaxe para criar um mapa de calor básico para este conjunto de dados:

 sns. heatmap (data)

Mapa de calor marítimo

Chá

Ajustar o tamanho do mapa de calor

Podemos usar o argumento figsize para ajustar o tamanho geral do mapa de calor:

 #set heatmap size
import matplotlib. pyplot as plt
plt. figure (figsize = (12.8))

#create heatmap
sns. heatmap (data) 

Mapa de calor marítimo ajusta tamanho

Alterar as cores do mapa de calor

Podemos usar o argumento cmap para alterar as cores usadas no mapa de calor. Por exemplo, poderíamos escolher a paleta de cores “Espectral”:

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

Mapa de calor Seaborn com argumento cmap

Ou poderíamos escolher a cartela de cores “coolwarm”:

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

Encontre uma lista completa de opções de cmap disponíveis aqui .

Anote o mapa de calor

Podemos usar a seguinte sintaxe para anotar cada célula no mapa de calor com formatação inteira e especificar o tamanho da fonte:

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

Mapa de calor marítimo com anotações

Alterar a barra de cores do mapa de calor

Finalmente, podemos desabilitar a barra de cores se quisermos usar o argumento cbar :

 sns. heatmap (data, cbar= False ) 

Mapa de calor Seaborn sem barra de cores

Encontre mais tutoriais do Seaborn nesta página .

Add a Comment

O seu endereço de email não será publicado. Campos obrigatórios marcados com *