]>
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 *
6 import matplotlib
.pyplot
as plt
8 def cree_figure(conf
,l_jours
,l_poids
,liste_err
):
13 jour_maxi
= max(l_jours
)# pas la peine d'aller très au delà du jour max
14 jour_maxi
= int(jour_maxi
* 1.1)+3 # on rajoute un peu
16 jour_maxi
= int(6*jours_dans_mois
) # si pas de données, arbitrairement on met ça
18 jour_maxi
= conf
["maxi"]+1
20 jour_maxi
= min(jour_maxi
,jours_maxi_courbe
)
22 if conf
["unite"] == "":
23 conf
["unite"] = choix_unite(jour_maxi
)
25 l_jours_conv
= convertit_tableau(l_jours
,conf
["unite"],liste_err
)
26 # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne...
27 age_maxi
= convertitunite(jour_maxi
-1,conf
["unite"],liste_err
)
30 titre
= "Courbe de poids OMS"
32 if conf
["typecourbe"] == "P":
34 liste_data_labels
= liste_data_labels_p
35 if conf
["sexe"] == "M":
36 fichier_oms
= f_poids_perc_garcon
37 titre
+= " (percentiles, garçon)"
38 elif conf
["sexe"] == "F":
39 fichier_oms
= f_poids_perc_fille
40 titre
+= " (percentiles, fille)"
42 fichier_oms
= f_poids_perc_mixte
43 titre
+= " (percentiles)"
44 elif conf
["typecourbe"] == "Z":
45 liste_data_labels
= liste_data_labels_z
46 if conf
["sexe"] == "G":
47 fichier_oms
= f_poids_z_garcon
48 titre
+= " (moyenne et écarts-types, garçon)"
49 elif conf
["sexe"] == "F":
50 fichier_oms
= f_poids_z_fille
51 titre
+= " (moyenne et écarts-types, fille)"
53 fichier_oms
= f_poids_z_mixte
54 titre
+= " (moyenne et écarts-types)"
56 # Si y'a un nom on met "courbe de machin"
58 titre
+= " de " +conf
["nom"]
61 # convertir les unités
63 #### On extrait les données des courbes, et on convertit les jours dans l'unité voulues
65 t
= lire_fichier_csv(fichier_oms
)
67 liste_err
[0].append("Impossible d'ouvrir le fichier "+fichier_oms
)
68 coljour
= convertit_tableau(extraire_colonne(t
,0,jour_maxi
),conf
["unite"],liste_err
)
72 fig
= plt
.figure(num
=None, figsize
=(conf
["largeur"], conf
["hauteur"]), dpi
=100, facecolor
='w', edgecolor
='k')
74 for (i
,label
,couleur
) in liste_data_labels
:
75 plt
.plot(coljour
,extraire_colonne(t
,i
,jour_maxi
),label
=label
,color
=couleur
)
77 # On extrait la valeur min et la valeur max des poids
78 (colonne_min
,_
,_
) = liste_data_labels
[0]
79 (colonne_max
,_
,_
) = liste_data_labels
[-1]
81 poids_min
= min(extraire_colonne(t
,colonne_min
,jour_maxi
))
82 poids_max
= max(extraire_colonne(t
,colonne_max
,jour_maxi
))
84 poids_min
= min(min(l_poids
),poids_min
)
85 poids_max
= max(max(l_poids
),poids_max
)
86 # On ajuste un peu ces min et max
87 # min : valeur min -1kg
88 poids_min
= max(0,poids_min
-1)
90 poids_max
= poids_max
* 1.05
93 plt
.plot(l_jours_conv
,l_poids
,label
=conf
["nom"],color
="black",marker
='o')
95 plt
.xlabel("Âge en "+conf
["unite"])
96 plt
.ylabel("Poids en kg")
98 plt
.axis([0,age_maxi
,poids_min
,poids_max
])
100 if conf
['legende']=="oui":
101 plt
.legend(loc
=conf
['positionlegende'])
102 plt
.grid(conf
["grille"]=="oui")