Aprenda Visualização de Dados Matriciais

Um heatmap é um gráfico onde os valores dos dados são representados como cores em uma matriz.

Esta é a forma padrão de visualizar matrizes de correlação (como as variáveis se relacionam entre si) ou grades de séries temporais (por exemplo, meses versus anos).

Importante: diferente de scatterplot ou barplot, que utilizam listas longas de dados, heatmap normalmente requer que seus dados estejam em formato de matriz (2D). Isso geralmente é obtido utilizando df.pivot_table() antes da plotagem.

Parâmetros Principais

annot=True: exibe o valor dos dados em cada célula;
cmap: o mapa de cores (gradiente) a ser utilizado. Opções comuns: 'viridis', 'coolwarm', 'magma';
fmt: código de formatação de string para controlar a aparência dos números;
'd': inteiros (sem casas decimais);
'.2f': floats com 2 casas decimais;
'g': formato geral (compacto);
linewidths / linecolor: adiciona bordas distintas entre as células.

Exemplo

Aqui está um heatmap mostrando a correlação entre variáveis numéricas no conjunto de dados tips.


              12345678910111213141516171819
            
import seaborn as sns
import matplotlib.pyplot as plt

# Load dataset
df = sns.load_dataset('tips')

# 1. Calculate Correlation Matrix (creates the grid)
corr_matrix = df.corr(numeric_only=True)

# 2. Plot Heatmap
sns.heatmap(
    data=corr_matrix,
    annot=True,     # Show numbers
    fmt='.2f',      # 2 decimal places
    cmap='coolwarm',# Red-Blue gradient
    linewidths=1    # Separation lines
)

plt.show()

Tarefa

Swipe to start coding

Visualizar o número de passageiros voando ao longo dos anos. Os dados já foram reorganizados em uma matriz (upd_df) para você utilizando pivot_table.

Definir o estilo como 'ticks'. Alterar a cor de fundo da figura para 'seagreen' ('figure.facecolor').
Criar um heatmap:

Utilizar upd_df como dados (este é o primeiro argumento, portanto não é necessário usar data=).
Utilizar o mapa de cores 'viridis' (cmap).
Exibir os números nas células (annot=True).
Formatar os números utilizando '0.99g' (formato geral).
Definir a cor das linhas entre as células como 'plum' (linecolor).

Exibir o gráfico.

Solução

Tudo estava claro?

Obrigado pelo seu feedback!

Seção 1. Capítulo 16

single

Pergunte à IA

Pergunte o que quiser ou experimente uma das perguntas sugeridas para iniciar nosso bate-papo

Deslize para mostrar o menu