Add more examples for exo1.

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

diff --git a/TP1/exo1/iris4.py b/TP1/exo1/iris4.py
new file mode 100755 (executable)
index 0000000..6646561
--- /dev/null
+++ b/TP1/exo1/iris4.py
@@ -0,0 +1,22 @@
+#!/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
+Y == 0
+X[Y == 0]
+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()