IntroductionToComputerScienceAndProgramming/Lec22.py at master · CyberSpaceXYZ/IntroductionToComputerScienceAndProgramming · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import random,pylab

fair=[1,2,3,4,5,6]

def throwPair(vals1,vals2):
    d1=random.choice(vals1)
    d2=random.choice(vals2)
    return d1,d2

def conductTrials(numThrows,die1,die2):
    throws=[]
    for i in range(numThrows):
        d1,d2=throwPair(die1,die2)
        throws.append(d1+d2)
    return throws

numThrows=100000

throws=conductTrials(numThrows,fair,fair)
pylab.hist(throws,11)
pylab.xticks(range(2,13),['2','3','4','5','6','7','8','9','10','11','12','13'])
pylab.title('Distribution of Values')
pylab.xlabel('Sum of Two Die')
pylab.ylabel('Number of Throws')

pylab.figure()
sums=pylab.array([0]*14)
for val in range(2,13):
    sums[val]=throws.count(val)
probs=sums[2:13]/float(numThrows)
xVals=pylab.arange(2,13)
pylab.plot(xVals,probs)
pylab.xticks(range(2,13),['2','3','4','5','6','7','8','9','10','11','12','13'])
pylab.title('Probability of Values')
pylab.xlabel('Sum of Two Die')
pylab.ylabel('Number of Throws')
pylab.figure()

def craps(die1,die2):
    d1,d2=throwPair(die1,die2)
    tot=d1+d2
    if tot in [7,11]: return True
    if tot in[2,3,12]: return False
    point=tot
    while True:
        d1,d2=throwPair(fair,fair)
        tot=d1+d2
        if tot==point: return True
        if tot==7: return False
def simCraps(numBets,die1,die2):
    wins,losses=[0,0]
    for i in range(numBets):
        if craps(die1,die2): wins+=1
        else: losses+=1
    print wins,losses
    houseWin=float(losses)/float(numBets)
    print houseWin
    print 'House winning percentage: '+str(houseWin)

simCraps(100000,fair,fair)
#simCraps(10000000,fair,fair)

weighted=[1,2,3,4,5,5,6] #good for gambler
throws=conductTrials(numThrows,fair,weighted)
sums=pylab.array([0]*14)
for val in range(2,13):
    sums[val]=throws.count(val)
probs=sums[2:13]/float(numThrows)
xVals=pylab.arange(2,13)
pylab.plot(xVals,probs)
pylab.legend()
simCraps(100000,fair,weighted)
pylab.show()