Finish TP1 exo1
[TP_AA.git] / TP1 / exo1 / TP1prog1.py
CommitLineData
ecf19406
JB
1#!/usr/bin/env python3
2
3# -*- coding: utf-8 -*-
4import pylab as pl
5
6from sklearn.datasets import load_iris
7irisData = load_iris()
8
9X = irisData.data
10Y = irisData.target
11
5b0cb811
JB
12colors = ["red", "green", "blue"]
13
ecf19406
JB
14x = 0
15y = 1
5b0cb811
JB
16for i in range(3):
17 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
18 label=irisData.target_names[i])
19pl.legend()
20pl.xlabel(irisData.feature_names[x])
21pl.ylabel(irisData.feature_names[y])
22pl.title(u"Données Iris - dimension des sépales uniquement")
23pl.show()
ecf19406 24
5b0cb811
JB
25x = 2
26y = 3
27for i in range(3):
28 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
29 label=irisData.target_names[i])
30pl.legend()
31pl.xlabel(irisData.feature_names[x])
32pl.ylabel(irisData.feature_names[y])
33pl.title(u"Données Iris - dimension des pétales uniquement")
34pl.show()
35
36x = 0
37y = 3
38for i in range(3):
39 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
40 label=irisData.target_names[i])
41pl.legend()
42pl.xlabel(irisData.feature_names[x])
43pl.ylabel(irisData.feature_names[y])
44pl.title(u"Données Iris - dimension des sépales et pétales")
45pl.show()
46
47x = 1
48y = 2
49for i in range(3):
50 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
51 label=irisData.target_names[i])
52pl.legend()
53pl.xlabel(irisData.feature_names[x])
54pl.ylabel(irisData.feature_names[y])
55pl.title(u"Données Iris - dimension des sépales et pétales")
56pl.show()
ecf19406 57
5b0cb811
JB
58x = 0
59y = 2
ecf19406
JB
60for i in range(3):
61 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
62 label=irisData.target_names[i])
5b0cb811
JB
63pl.legend()
64pl.xlabel(irisData.feature_names[x])
65pl.ylabel(irisData.feature_names[y])
66pl.title(u"Données Iris - dimension des sépales et pétales")
67pl.show()
ecf19406 68
5b0cb811
JB
69x = 1
70y = 3
71for i in range(3):
72 pl.scatter(X[Y == i][:, x], X[Y == i][:, y], color=colors[i],
73 label=irisData.target_names[i])
ecf19406
JB
74pl.legend()
75pl.xlabel(irisData.feature_names[x])
76pl.ylabel(irisData.feature_names[y])
5b0cb811 77pl.title(u"Données Iris - dimension des sépales et pétales")
ecf19406 78pl.show()