import random
from matplotlib import pyplot

def streak(p):
    L=[0]
    M = 0
    for m in range (1, 201):
        if random.random()<p:
            M = M + 1
        else:
            M = M -1
        L.append(M)
    print (L)
    pyplot.plot(L)
    pyplot.title('Wins and losses vs. number of investments')
    pyplot.xlabel('Number of investments')
    pyplot.ylabel('Wins or losses')
    pyplot.show()
    
def fixed(p): # p is the probability of winning
    bank=1000
    L=[bank]
    m = 0
    while bank > 0 and m < 200:
        if random.random()<p:
            bank = bank+200 #wagering fixed amount
        else:
            bank = bank-200
        L.append(bank)
        m = m+1
    print (L)
    print ("total", bank, "max", max(L), "min", min(L))
    pyplot.plot(L)   
    pyplot.title('Bankroll vs. number of investments, fixed investments')
    pyplot.xlabel('Number of investments')
    pyplot.ylabel('Bankroll')
    pyplot.show()

def kelly(p): # p is the probability of winning
    bank=1000
    L=[bank]
    m= 0
    while bank > 0 and m < 200:
        if random.random()<p:
            bank = bank*(2*p) #wagering using Kelly criterion
        else:
            bank = bank*(2-2*p)
        L.append(int(bank))
        m = m + 1
    print (L)
    print ("total", int(bank), "max", int(max(L)), "min", int(min(L)))
    pyplot.plot(L)   
    pyplot.title('Bankroll vs. number of investments, using Kelly criterion')
    pyplot.xlabel('Number of investments')
    pyplot.ylabel('Bankroll')
    pyplot.show()