Skip to content

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.

plt.figure(figsize=(6,5))
corr = df[['math score', 'reading score', 'writing score']].corr()
sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
plt.title('Heatmap de Correlação das Notas')
plt.tight_layout()
plt.close()