[Treinando]

import numpy as np
import matplotlib.pyplot as plt

# Carregando os dados do arquivo (substitua 'dados.csv' pelo nome real do seu arquivo)
# Supondo que o separador seja vírgula e que não tenha cabeçalho
dados = np.loadtxt('dados.csv', delimiter=',')

# Seleção dos dados
# Diâmetro está na coluna 0 e peso na coluna 1

# Laranjas: da linha 0 até a 4999
diametros_laranja = dados[0:5000, 0]
pesos_laranja = dados[0:5000, 1]

# Toranjas: da linha 5000 em diante
diametros_toranja = dados[5000:, 0]
pesos_toranja = dados[5000:, 1]

# Criando o gráfico
plt.figure(figsize=(10, 6))
plt.scatter(diametros_laranja, pesos_laranja, color='orange', label='Laranja', alpha=0.5)
plt.scatter(diametros_toranja, pesos_toranja, color='red', label='Toranja', alpha=0.5)

plt.xlabel('Diâmetro (cm)')
plt.ylabel('Peso (g)')
plt.title('Peso x Diâmetro - Laranja vs Toranja')
plt.legend()
plt.grid(True)
plt.show()