]>
git.immae.eu Git - perso/Denise/oms.git/blob - trace_courbe.py
2 # -*- coding: utf-8 -*-
3 from configuration
import *
4 from gestionOMS
import *
5 from gestion_unites
import *
7 import matplotlib
.pyplot
as plt
9 def cree_figure(conf
,l_jours
,l_poids
,liste_err
):
10 #warning("debut de cree_figure"+str(conf),liste_err)
12 liste_data_labels_p
,liste_data_labels_z
= renvoie_liste_labels(conf
,liste_data_choisie_p
,liste_data_choisie_z
,liste_err
)
14 erreur("bug avec liste data labels",liste_err
)
17 #warning("bla",liste_err)
20 jour_maxi
= max(l_jours
)# pas la peine d'aller très au delà du jour max
21 jour_maxi
= int(jour_maxi
* 1.1)+3 # on rajoute un peu
23 jour_maxi
= int(6*jours_dans_mois
) # si pas de données, arbitrairement on met ça
25 jour_maxi
= conf
["maxi"]+1
27 jour_maxi
= min(jour_maxi
,jours_maxi_courbe
)
29 if conf
["unite"] == "":
30 conf
["unite"] = choix_unite(jour_maxi
)
32 l_jours_conv
= convertit_tableau(l_jours
,conf
["unite"],liste_err
)
33 # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne...
34 age_maxi
= convertitunite(jour_maxi
-1,conf
["unite"],liste_err
)
37 titre
= "Courbe de poids OMS"
39 # warning("cree_figure : ça va jusque là ",liste_err)
40 #warning("bli"+str(conf),liste_err)
43 if conf
["typecourbe"] == "P":
45 liste_data_labels
= liste_data_labels_p
46 if conf
["sexe"] == "M":
47 fichier_oms
= f_poids_perc_garcon
48 titre
+= " (percentiles, garçon)"
49 elif conf
["sexe"] == "F":
50 fichier_oms
= f_poids_perc_fille
51 titre
+= " (percentiles, fille)"
53 fichier_oms
= f_poids_perc_mixte
54 titre
+= " (percentiles)"
55 elif conf
["typecourbe"] == "Z":
56 liste_data_labels
= liste_data_labels_z
57 if conf
["sexe"] == "G":
58 fichier_oms
= f_poids_z_garcon
59 titre
+= " (moyenne et écarts-types, garçon)"
60 elif conf
["sexe"] == "F":
61 fichier_oms
= f_poids_z_fille
62 titre
+= " (moyenne et écarts-types, fille)"
64 fichier_oms
= f_poids_z_mixte
65 titre
+= " (moyenne et écarts-types)"
67 warning("Type de courbe invalide"+conf
["typecourbe"],liste_err
)
69 # warning("cree_figure : ça va jusque là ",liste_err)
70 # warning("bli"+str(conf),liste_err)
73 # Si y'a un nom on met "courbe de machin"
75 titre
+= " de " +conf
["nom"]
78 # convertir les unités
80 #### On extrait les données des courbes, et on convertit les jours dans l'unité voulues
82 t
= lire_fichier_csv(fichier_oms
)
84 liste_err
[0].append("Impossible d'ouvrir le fichier "+fichier_oms
)
85 coljour
= convertit_tableau(extraire_colonne(t
,0,jour_maxi
),conf
["unite"],liste_err
)
89 fig
= plt
.figure(num
=None, figsize
=(conf
["largeur"], conf
["hauteur"]), dpi
=100, facecolor
=conf
["couleurs"]["fond"])
90 plt
.rcParams
['axes.facecolor'] = conf
["couleurs"]["fond"]
91 plt
.rcParams
['axes.edgecolor']= conf
["couleurs"]["cadretxt"]
92 plt
.rcParams
['xtick.color'] = conf
["couleurs"]["cadretxt"]
93 plt
.rcParams
['ytick.color'] = conf
["couleurs"]["cadretxt"]
94 plt
.rcParams
['grid.color'] = conf
["couleurs"]["grille"]
95 plt
.rcParams
['legend.edgecolor'] = conf
["couleurs"]["grille"]
97 #warning("bla"+str(liste_data_labels),liste_err)
98 for (i
,label
,couleur
) in liste_data_labels
:
99 plt
.plot(coljour
,extraire_colonne(t
,i
,jour_maxi
),label
=label
,color
=couleur
)
100 #warning("colonne"+str(i)+str(label)+str(couleur),liste_err)
104 # On extrait la valeur min et la valeur max des poids
105 (colonne_min
,_
,_
) = liste_data_labels
[-1]
106 (colonne_max
,_
,_
) = liste_data_labels
[0]
108 poids_min
= min(extraire_colonne(t
,colonne_min
,jour_maxi
))
109 poids_max
= max(extraire_colonne(t
,colonne_max
,jour_maxi
))
111 poids_min
= min(min(l_poids
),poids_min
)
112 poids_max
= max(max(l_poids
),poids_max
)
113 # On ajuste un peu ces min et max
114 # min : valeur min -1kg
115 poids_min
= max(0,poids_min
-1)
117 poids_max
= poids_max
* 1.05
119 #warning("Jusque là ça marche",liste_err)
122 plt
.plot(l_jours_conv
,l_poids
,label
=conf
["nom"],color
=conf
["couleur_cadretxt"],marker
='o')
124 plt
.xlabel("Âge en "+conf
["unite"],color
=conf
["couleurs"]["cadretxt"])
125 plt
.ylabel("Poids en kg",color
=conf
["couleurs"]["cadretxt"])
126 plt
.title(titre
,color
=conf
["couleurs"]["cadretxt"])
127 plt
.axis([0,age_maxi
,poids_min
,poids_max
])
129 if conf
['legende']=="oui":
130 legende
= plt
.legend(loc
=conf
['positionlegende'])
131 plt
.setp(legende
.get_texts(), color
=conf
["couleurs"]["cadretxt"])
132 plt
.grid(conf
["grille"]=="oui")
136 #warning("Jusque là ça marche",liste_err)