# Créé par eclermont, le 21/02/2023 en Python 3.7
from sklearn.datasets import load_iris
import numpy as np
iris = load_iris()
import matplotlib.pyplot as plt

X1_setosa =[]
X2_setosa = []
X1_versicolor= []
X2_versicolor= []
X1_virginica = []
X2_virginica = []

# Création des listes avec les 2 1eres caracteristiques de chaque classe (0,1,2):
for i in range( len(iris.target) ): # recupération de chaque iris
  liris = iris.data[i]
  classe = iris.target[i]

  if classe == 0 :
    X1_setosa.append(liris[0])
    X2_setosa.append(liris[1])
  elif classe ==1:
    X1_versicolor.append(liris[0])
    X2_versicolor.append(liris[1])
  else :
    X1_virginica.append(liris[0])
    X2_virginica.append(liris[1])

# Tracé du graphique:
plt.title('Espèces d iris')
plt.xlabel('Longueur de sepale')  # titre de l'axe des abscisses
plt.ylabel('Largeur de sepale')  # titre de l'axe des ordonnées


plt.scatter(X1_setosa,X2_setosa, color='g', label='setosa')
plt.scatter(X1_versicolor,X2_versicolor, color='r', label='versicolor')
plt.scatter(X1_virginica,X2_virginica, color='b', label='Virginica')
plt.legend()
plt.show()