stats_longueur.py

import matplotlib.pyplot as plt
from Cube import *
import json
from lire_entree import lecture_cube
from stats import moyenne
from algo import *

"""
Génération de stats sur la longueur de la solution
"""

JEU_TEST = 'tests/samples/liste-sample.json'

with open(JEU_TEST) as data_file: #on parse le jeu de test JSON
        data = json.load(data_file)
        tests = data["cubes"]
        # dictionnaire pour stocker en clé : nbMouvement et
        #en valeur : Occurence sur les n tests
        listeNbMouvements = {}
        for test in tests: # on parcours tout les cubes
            b,c = lecture_cube(test)
            c,mouv = algo_cfop(c) # on fais l'algo
            # si le nombre de mouvements est deja dans la dictionnaire,
            # on ajoute 1 à son occurence
            if len(mouv) in listeNbMouvements:
                listeNbMouvements[len(mouv)] += 1
            else:
                listeNbMouvements[len(mouv)] = 1 # sinon on l'ajoute

listeX = []
listeY = []
# on "separe" le dictionnaire en deux liste correspondant aux clés et aux valeurs
for x,y in listeNbMouvements.items():
    listeX.append(x)
    listeY.append(y)

plt.plot(listeX,listeY) # affichage de la courbe dans pyplot
plt.xlabel('Nombre de mouvements')
plt.ylabel('Occurence sur '+str(len(tests))+' test')
plt.show()