Add files via upload

main.py

@@ -0,0 +1,11 @@
+# --imports
+import matplotlib.pyplot as plt
+# --class imports
+from roulette import Roulette
+# --run
+
+r = Roulette(10000)
+r.plot_stats()
+plt.figure('Banque')
+plt.plot(r.martingale_simple('Noir', 10))
+plt.show()

roulette.py

@@ -0,0 +1,171 @@
+
+# --imports
+import matplotlib.pyplot as plt
+import numpy as np
+from numpy.random import randint
+
+# --class
+class Roulette(object):
+    def __init__(self, k):
+        self.nb_tirages = k
+        self.tirages = self.__generer_tirages(self.nb_tirages)
+
+        line = k*['        ']
+        self.results = np.array([line,line,line,line])
+
+        self.__pair_impair()
+        self.__couleur()
+        self.__passe_manque()
+        self.__colonne()
+        self.stats = {}
+        self.__compute_stats()
+
+        self.COLORS = ['g','r','k','r','k','r','k','r','k','r','k','k','r','k','r','k','r','k','r','r','k','r','k','r','k','r','k','r','k','k','r','k','r','k','r','k','r']
+        self.NUMBERS = [0,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]
+    # --getters
+    def get_results(self):
+        return self.results
+
+    # --private methods
+    def __generer_tirages(self, i):
+        return randint(0,37,i)
+
+    def __compute_stats(self):
+        self.stats['Parite'] = self.to_stat(self.results[0,:], ['Pair', 'Impair', 'Zero'])
+        self.stats['Couleur'] = self.to_stat(self.results[1,:], ['Rouge', 'Noir', 'Zero'])
+        self.stats['PM'] = self.to_stat(self.results[2,:], ['Passe', 'Manque', 'Zero'])
+        self.stats['Colonne'] = self.to_stat(self.results[3,:], ['Premiere', 'Milieu', 'Derniere', 'Zero'])
+
+    def __pair_impair(self):
+        for i in xrange(len(self.tirages)):
+            if self.tirages[i]%2 == 0 and self.tirages[i] != 0 :
+                self.results[0,i] = 'Pair'
+            elif self.tirages[i] == 0 :
+                self.results[0,i] = 'Zero'
+            else :
+                self.results[0,i] = 'Impair'
+
+    def __couleur(self):
+        red = [1, 3, 5, 7, 9, 12, 14, 16, 18, 19, 21, 23, 25, 27, 30, 32, 34, 36]
+        for i in xrange(len(self.tirages)):
+            if self.tirages[i] in red :
+                self.results[1,i] = 'Rouge'
+            elif self.tirages[i] == 0 :
+                self.results[1,i] = 'Zero'
+            else :
+                self.results[1,i] = 'Noir'
+
+    def __passe_manque(self):
+        for i in xrange(len(self.tirages)):
+            if self.tirages[i]<19 and self.tirages[i]>0:
+                self.results[2,i] = 'Passe'
+            elif self.tirages[i] == 0 :
+                self.results[2,i] = 'Zero'
+            else :
+                self.results[2,i] = 'Manque'
+
+    def __colonne(self):
+        for i in xrange(len(self.tirages)):
+            if self.tirages[i]%3 == 1 and self.tirages[i] != 0:
+                self.results[3,i] = 'Premiere'
+            elif self.tirages[i] == 0 :
+                self.results[3,i] = 'Zero'
+            elif self.tirages[i]%3 == 2 :
+                self.results[3,i] = 'Milieu'
+            else :
+                self.results[3,i] = 'Derniere'
+
+    # --public methods
+    def plot_stats(self):
+        plt.figure('Statistics')
+        plt.subplot(241)
+        barlist = plt.bar(range(len(self.stats['Parite'])), self.stats['Parite'].values(), align='center')
+        barlist[2].set_color('g')
+        plt.xticks(range(len(self.stats['Parite'])), self.stats['Parite'].keys())
+        plt.subplot(242)
+        barlist = plt.bar(range(len(self.stats['Couleur'])), self.stats['Couleur'].values(), align='center')
+        barlist[0].set_color('g')
+        barlist[1].set_color('k')
+        barlist[2].set_color('r')
+        plt.xticks(range(len(self.stats['Couleur'])), self.stats['Couleur'].keys())
+        plt.subplot(245)
+        barlist = plt.bar(range(len(self.stats['PM'])), self.stats['PM'].values(), align='center')
+        barlist[1].set_color('g')
+        plt.xticks(range(len(self.stats['PM'])), self.stats['PM'].keys())
+        plt.subplot(246)
+        barlist = plt.bar(range(len(self.stats['Colonne'])), self.stats['Colonne'].values(), align='center')
+        barlist[2].set_color('g')
+        plt.xticks(range(len(self.stats['Colonne'])), self.stats['Colonne'].keys()) 
+        plt.subplot(122)
+
+        labels = self.text_to_stat(self.lnum_to_lstr(self.tirages)).keys()
+        sizes = self.text_to_stat(self.tirages).values()
+        colors = self.COLORS
+        plt.pie(sizes, labels=labels, shadow=True, colors=colors)
+
+    def display_roulette(self):
+        pass
+
+    def martingale_simple(self, mise_initiale, banque_initiale):
+        cl = self.results[1,:]
+        mise = []
+        gagnant = [] 
+        gains = []
+        banque = banque_initiale
+        banque_max = banque_initiale
+        banque_vec = []
+        banque_vec.append(banque_initiale)
+        enchere_courante = 1
+        mise.append(mise_initiale)
+        for k in xrange(len(cl)):
+            if mise[k] == cl[k]:
+                gain = enchere_courante*2
+                enchere_courante = 1
+                gagnant.append('Oui')
+                if mise[k]=='Rouge':
+                    mise.append('Noir')
+                else :
+                    mise.append('Rouge')
+            else :
+                gain = -enchere_courante
+                enchere_courante = enchere_courante*2
+                gagnant.append('Non')
+                mise.append(mise[k])
+            gains.append(gain)
+            banque += gain
+            if banque > banque_max :
+                banque_max = banque
+            if banque<0 :
+                print 'Perdu'
+                print 'gains', gains
+                print 'banque maximale =', banque_max
+                return banque_vec
+            banque_vec.append(banque)
+        print 'gains', gains
+        print 'banque maximale =', banque_max 
+        return banque_vec
+
+    def to_stat(self, l, l_keys):
+        d = {}
+        for k in l_keys:
+            d[k]=0
+        for i in xrange(len(l)):
+            if not l[i] in l_keys:
+                print 'Error', l[i], 'is not in the pre-defined keys'
+            else :
+                d[l[i]] += 1
+        return d
+
+    def text_to_stat(self, l):
+        d = {}
+        for i in xrange(len(l)):
+            if not l[i] in d.keys():
+                d[l[i]] = 1
+            else :
+                d[l[i]] += 1
+        return d
+
+    def lnum_to_lstr(self, l):
+        for i,x in enumerate(l):
+            l[i] = str(x)
+        return l