import matplotlib.pyplot as plt
import matplotlib.colors as col
import numpy as np
import pandas as pd
import sys

iris_df = pd.read_csv('iris.csv')


#for k,v in col.CSS4_COLORS.items():
#	print(k, ' ', v)

colors = [ col.CSS4_COLORS['goldenrod'], col.CSS4_COLORS['yellowgreen'], col.CSS4_COLORS['maroon'] ]
species = ['Setosa', 'Versicolor', 'Virginica']

i = 0
for s in species:
	df = iris_df[ iris_df.variety == s ]
	# prendre sepal.length pour x_values
	x_values = df.iloc[:,0]
	# prendre sepal width pour y_values
	y_values = df.iloc[:,1]
	
	plt.scatter(x_values, y_values, color=colors[i])
	i = i + 1

plt.xlabel('longueur sépale')
plt.ylabel('largeur sépale')
plt.savefig('img/plot_scatter_iris_sepal.png')
plt.show()