This repository has been archived by the owner on Feb 4, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrepresentation.py
78 lines (74 loc) · 2.78 KB
/
representation.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import matplotlib.pyplot as plt
def ploteur2d(X,Y,Z,titreX="titreX",titreY="titreY",colorbar_nom="colorbar",nom="figure",save=False):
"""
Description : Realise un trace contour à partir des listes fournies.
---
variables d'entree :
X : tableau contenant les positions x du domaine à representer.
Y : tableau contenant les positions y du domaine à representer.
Z : tableau contenant les valeurs z du domaine à representer.
titreX : titre de l'axe x. "titreX" par defaut.
titreY : titre de l'axe y. "titreY" par defaut.
nom : non de l'image sous laquelle elle sera enregistree. "figure" par defaut.
save : booleen. False par defaut.
---
variables de sortie :
Si save est False, affiche une figure .
Si save est True, enregistre l'image au nom de la variable nom.
"""
plt.figure()
plt.contourf(X,Y,Z,1000)
plt.xlabel(titreX)
plt.ylabel(titreY)
cbar=plt.colorbar()
cbar.set_label(colorbar_nom)
if not save:
plt.show()
else :
plt.savefig("Pictures/"+nom)
def ploteur(X,Y,labels,format,titreX="titreX",titreY="titreY",nom="figure",save=False,legend=True):
"""
Description : Realise un trace à partir des listes fournies.
---
variables d'entree :
X : tableau contenant les positions x du domaine à representer. Chaque ligne correspond à un jeu de donnees.
Y : tableau contenant les positions y du domaine à representer. Chaque ligne correspond à un jeu de donnees.
titreX : titre de l'axe x. "titreX" par defaut.
titreY : titre de l'axe y. "titreY" par defaut.
nom : non de l'image sous laquelle elle sera enregistree. "figure" par defaut.
save : booleen. False par defaut.
---
variables renvoyees :
Si save est False, affiche une figure.
Si save est True, enregistre l'image au nom de la variable nom.
"""
n=len(X)
plt.figure()
for i in range(n):
plt.plot(X[i],Y[i],format[i],label=labels[i])
plt.grid()
if legend:
plt.legend()
plt.xlabel(titreX)
plt.ylabel(titreY)
if not save:
plt.show()
else :
plt.savefig("Pictures/"+nom)
def histogramme(X,N,Xlabel,nom):
"""
Description : Fonction dessinant l'histogramme d'une liste X pour N intervalles
---
variables d'entree :
X : Liste contenant les valeurs à afficher dans l'histogramme
N : Entier contenant le nombre d'intervalle de l'histogramme
nom : Nom de la figure dans les dossier
---
variables renvoyees :
L'image de l'histogramme est stockee sous format .png dans le dossier 'data'.
"""
plt.figure()
plt.hist(X, range = (min(X),max(X)) , bins = (N))
plt.xlabel(Xlabel)
plt.ylabel("Nombre de simulations")
plt.savefig("Pictures/"+ nom)