5. Correlação entre Variáveis
Nesta etapa, é apresentado o heatmap de correlação entre as notas dos estudantes, permitindo visualizar as relações entre matemática, leitura e escrita.
import matplotlib.pyplot as plt
import seaborn as sns
import os
# Calcula a matriz de correlação entre as notas
corr = df[['math score', 'reading score', 'writing score']].corr()
# Cria o diretório de imagens se não existir
os.makedirs('imagens', exist_ok=True)
# Gera o heatmap
plt.figure(figsize=(6,5))
sns.heatmap(corr, annot=True, cmap='coolwarm')
plt.title('Heatmap de Correlação entre Notas')
plt.savefig('imagens/heatmap_correlacao_notas.png')
plt.show()
