Finish TP1 exo1

[TP_AA.git] / TP1 / exo1 / iris4.py

#!/usr/bin/env python3

import pylab as pl  # permet de remplacer le nom "pylab" par "pl"

from sklearn.datasets import load_iris  # les données iris sont chargées
irisData = load_iris()

X = irisData.data
Y = irisData.target
x = 0
y = 1
print(Y == 0)
print(X[Y == 0])
print(X[Y == 0][:, x])
pl.scatter(X[Y == 0][:, x], X[Y == 0][:, y],
           color="red", label=irisData.target_names[0])
pl.scatter(X[Y == 1][:, x], X[Y == 1][:, y],
           color="green", label=irisData.target_names[1])
pl.scatter(X[Y == 2][:, x], X[Y == 2][:, y],
           color="blue", label=irisData.target_names[2])
pl.legend()
pl.show()