91 lines
2.0 KiB
Python
91 lines
2.0 KiB
Python
|
|
|
|
import matplotlib.pyplot as plt; plt.rcdefaults()
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
from pylab import *
|
|
|
|
|
|
TailleGrilleX = 15
|
|
TailleGrilleY = 15
|
|
GrilleInitiale = 0
|
|
HistoriqueGrille = list()
|
|
StratsResultats = list()
|
|
ListeStrategies = list()
|
|
|
|
|
|
def recup_mat(matrice, param):
|
|
array*str-> array
|
|
Récupère la matrice avec seulement le paramètre (état,strat,années de prison) voulu afin de pouvoir avoir les stats plus facilement"""
|
|
|
|
|
|
matR = np.random.randint(0,1,TailleGrilleX,TailleGrilleY)
|
|
|
|
for ligne in range (0,TailleGrilleX): #int ligne
|
|
for colonne in range (0, TailleGrilleY): #int colonne
|
|
matR[ligne][colonne]=Grille [ligne][colonne][param]
|
|
|
|
return matR
|
|
|
|
def bar_chart():
|
|
"""array->graph
|
|
Retourne les diagrammes en baton qui mettent en évidence le nombre d'années
|
|
de prison en fonction de la stratégie et le nombre d'utilisation de chaque stratégies
|
|
|
|
|
|
matStrat=recup_mat('strategie')
|
|
matGain= recup_mat('annees_de_prison')
|
|
|
|
#initialisation des paramètres
|
|
#list gain
|
|
gain=[]
|
|
#list strat
|
|
strat=[]
|
|
|
|
for i in range(5):
|
|
|
|
gain.append(0)
|
|
strat.append(0)
|
|
|
|
|
|
|
|
|
|
|
|
for i in range(0,len(ListeStrategies)):
|
|
strat[i]=StratsResultats[i]['nb_utilisations']
|
|
gain[i]=StratsResultats[i]['total_ans_prisons']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Strat=('1','2', '3','4' ,'5')
|
|
|
|
x_pos = np.arange(len(Strat))
|
|
|
|
plt.subplot(221)
|
|
|
|
plt.bar(x_pos, gain, align='center', color='b' )
|
|
|
|
plt.xlabel("Stratégies")
|
|
plt.ylabel("Nombre d'années de prison")
|
|
|
|
plt.subplot(222)
|
|
plt.bar(x_pos, strat, align='center' , color='r')
|
|
plt.xlabel("Stratégies")
|
|
plt.ylabel("NB individus ayant adopté stratégie")
|
|
|
|
|
|
plt.xticks(x_pos,Strat)
|
|
|
|
plt.show()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|