ARE_Dynamic_2017/parties/Bar_Chart + Récup_mat.py

103 lines
2.2 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 simulation():
"""->Grille"""
iterations = 0
while iterations <= MaxIterations :
partie_globale()
pass
def matStratIté("strategie", i):
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)
matrice = HistoriqueGrille[i]
for ligne in range (0,TailleGrilleX): #int ligne
for colonne in range (0, TailleGrilleY): #int colonne
matStrat[ligne][colonne]=matrice[ligne][colonne]["strategie"]
return matStrat
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()