# Import der Zufallsbibliothek und zweier Bibliotheken für Mathematikanwendungen
import random
import matplotlib.pyplot as plt
import numpy as np

# Anzahl der durchzuführenden Durchgänge
ANZAHL_DURCHGAENGE = 50

# Anzahl der Spielautomaten
anzahl_banditen = 10


# Deklariere eine Listen-Variable namens "summe", 
# die die Gesamtzahl aller Gewinne für jeden der Banditen zählt und setze sie auf Null.
summe = np.zeros(10)
# Nach jedem Durchgang werden die Werte der Liste "summe" gespeichert, zuerst werden die Einträge dieser Matrix auf 0 gesetzt.
status = np.zeros((anzahl_banditen,ANZAHL_DURCHGAENGE))
# Deklariere eine Variable "weights", welche für jeden Spielautomaten eine Liste mit der Verlier- und Gewinnwahrscheinlichkeit enthält. 
weights = [[0.9,0.1],[0.5,0.5],[0.4,0.6],[0.2,0.8],[0.9,0.1],[0.75,0.25],[0.4,0.6],[0.55,0.45],[0.25,0.75],[0.35,0.65]]

# Beginn einer for-Schleife, die so oft ausgeführt wird,
# wie die Variable ANZAHL_DURCHGAENGE angibt.
# Nenne die Laufvariable "zug".
for zug in range(ANZAHL_DURCHGAENGE):
    # In einem Durchgang wird mit jedem Banditen einmal gespielt
    for j in range(anzahl_banditen):
        aktueller_wurf = random.choices([0,1],weights[j])
        summe[j] += aktueller_wurf
        status[j,zug] =  summe[j]


# Labels, Titel, Legende werden geplottet und am Ende mit plt.show() wird der Plot angezeigt

plt.plot(status[0,:], label= 'A', color = (0,162/255,255/255))
plt.plot(status[1,:], label= 'B', color = (97/255,216/255,54/255))
plt.plot(status[2,:], label= 'C', color = (238/255,34/255,12/255))
plt.plot(status[3,:], label= 'D', color = (0, 0, 128/255))
plt.plot(status[4,:], label= 'E', color = (0, 128/255, 128/255))
plt.plot(status[5,:], label= 'F', color = (128/255, 0, 128/255))
plt.plot(status[6,:], label= 'G', color = (0, 128/255, 0))
plt.plot(status[7,:], label= 'H', color = (128/255, 128/255, 0))
plt.plot(status[8,:], label= 'I', color = (128/255, 0, 0))
plt.plot(status[9,:], label= 'J', color = (128/255, 128/255, 128/255))

plt.ylabel('Anzahl Treffer')
plt.xlabel('Anzahl Druchgänge')
plt.title('Gauner-Übersicht')
plt.legend()
plt.show()


print(plt.style.available)