import pandas as pd
from scipy.stats import pearsonr, spearmanr
dados = pd.DataFrame({
'horas_estudadas': [1, 2, 3, 4, 5],
'notas': [6, 6.5, 7, 7.5, 8]
})
dados_mod = pd.DataFrame({
'horas_estudadas': [1, 2, 3, 4, 5],
'notas': [6, 6.5, 6.8, 6.95, 7]
})
# Calculando a correlação de Pearson para DataFrame "dados"
coef_pearson, p_valor = pearsonr(dados['horas_estudadas'], dados['notas'])
print(coef_pearson)
# output: 1.0
# Calculando a correlação de Pearson para DataFrame "dados_mod"
coef_pearson, p_valor = pearsonr(dados_mod['horas_estudadas'], dados_mod['notas'])
print(coef_pearson)
# output: 0.9395
# Calculando a correlação de Spearman para DataFrame "dados"
coef_spearman, p_valor = spearmanr(dados['horas_estudadas'], dados['notas'])
print(coef_spearman)
# output: 0.99999999
# Calculando a correlação de Spearman para DataFrame "dados_mod"
coef_spearman, p_valor = spearmanr(dados_mod['horas_estudadas'], dados_mod['notas'])
print(coef_spearman)
# output: 0.99999999