Untitled

premio_nobel = np.array([405,137,114,72,33,32,29,28,27,23])
consumo_chocolate = np.array([4.31,7.39,7.89,4.22,5.4,5.3,3.56,4.89,8.98,7.8])
print(premio_nobel) # x
print(consumo_chocolate) # y

# Pega os dados (Só um exemplo)
x = premio_nobel # Variável X
y = consumo_chocolate # Vriável Y

# plt é o objeto que contém as funções para editar
# e exibir a figura
plt.title("Prêmios Nobel X Consumo de Chocolate (Per capita (kg)") # Define o título do gráfico
plt.xlabel("Prêmios Nobel") # Define a legenda para o eixo X
plt.ylabel("Consumo de Chocolate") # Define a legenda para o eixo Y

# 'Plota' (exibe) o gráfico passando x, y e o estilo dos pontos
plt.plot(x,y, 'ob') # OB significa O é um circulo e B é a cor blue (azul)

# Realiza o cálculo do MMQ
equação = np.polyfit(x, y, 1)

# Adicionar a linha de tendência
linha_tendência = np.poly1d(equação) # Realiza o cálculo da linha de tendência

# Calcula a correlação de Pearson
correlação = np.corrcoef(x,y)[0][1] 

# Adiciona a linha na figura
plt.plot(x, linha_tendência(x), 'r--')

# Definindo a legenda para exibir na figura
legenda = f'Linha de tendência: y = {equação[0]:.6f}x {equação[1]:.6f}'
legenda += f'\nCoeficiente de Pearson (p): {correlação}'
plt.text(0.05, 0.9, legenda, 
         transform=plt.gca().transAxes, 
         fontsize=11)
#-------------------------------------------

# Exibe a figura completa
plt.show()