import numpy as np
import matplotlib.pyplot as plt

# import pandas as pd

p = open(r"\\SERVEUR01\HOSPITF\Travail\Téléchargements\data/penguins.csv") #mettre votre chemin
penguins={}
l=p.readline()
name=l.strip().split(',')
for i in name:
    penguins[i]=[]

t=p.readlines()

for i in range(len(t)) :
    L=t[i].strip().split(',')
    penguins[name[0]].append(L[0])
    penguins[name[1]].append(L[1])
    penguins[name[6]].append(L[6])
    penguins[name[7]].append(int(L[7]))
    for j in range(2,6):
        penguins[name[j]].append(float(L[j]))


print(penguins)
##
fig = plt.figure(figsize=(15, 5))
# Figure 1 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 1)
ax.scatter(penguins["bill_length_mm"], penguins["body_mass_g"])
plt.xlabel("Longueur du bec (mm)")
plt.ylabel("Poids (g)")
# Figure 2 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 2)
ax.scatter(penguins["bill_depth_mm"], penguins["body_mass_g"])
plt.xlabel("Hauteur du bec (mm)")
plt.ylabel("Poids (g)")
# Figure 3 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 3)
ax.scatter(penguins["flipper_length_mm"], penguins["body_mass_g"])
plt.xlabel("Longueur de la palette natatoire (mm)")
plt.ylabel("Poids (g)")


X = np.array( penguins[["bill_length_mm", "bill_depth_mm", "flipper_length_mm"]])
y_regress = np.array( penguins["body_mass_g"])

##
plt.ion()
fig = plt.figure(figsize=(15, 5))
coul={'male':'b','female':'g'}
# Figure 1 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 1)
for penguin_sex in ["male", "female"]:
    ax.scatter(penguins.loc[penguins["sex"] == penguin_sex]["bill_length_mm"],penguins.loc[penguins["sex"] == penguin_sex]["bill_depth_mm"],c=coul[penguin_sex],label=penguin_sex)
plt.xlabel("Longueur du bec (mm)")
plt.ylabel("Hauteur du bec (mm)")
# Figure 2 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 2)
for penguin_sex in ["male", "female"]:
    ax.scatter(penguins.loc[penguins["sex"] == penguin_sex]["bill_length_mm"],penguins.loc[penguins["sex"] == penguin_sex]["flipper_length_mm"],c=coul[penguin_sex],label=penguin_sex)
plt.xlabel("Longueur du bec (mm)")
plt.ylabel("Longueur de la palette natatoire (mm)")
# Figure 3 dans une grille (1 x 3)
ax = plt.subplot(1, 3, 3)
for penguin_sex in ["male", "female"]:
    ax.scatter(penguins.loc[penguins["sex"] == penguin_sex]["bill_depth_mm"], penguins.loc[penguins["sex"] == penguin_sex]["flipper_length_mm"],c=coul[penguin_sex],label=penguin_sex)
plt.xlabel("Hauteur du bec (mm)")
plt.ylabel("Longueur de la palette natatoire (mm)")
plt.legend()