# CS 237 Spring 2019
# Author: Alina Ene (aene@bu.edu)
# Used in L6
import numpy as np
from numpy.random import randint
import matplotlib.pyplot as plt
plt.style.use('seaborn')
blue_die = [0, 0, 4, 4, 4, 4]
green_die = [1, 1, 1, 5, 5, 5]
orange_die = [2, 2, 2, 2, 6, 6]
yellow_die = [3, 3, 3, 3, 3, 3]
# die1 vs die2
# in each roll, the larger number wins
def roll_dice(die1, die2, num_rolls = 10000):
die1_wins = 0
for i in range(num_rolls):
n1 = die1[randint(1, 7) - 1]
n2 = die2[randint(1, 7) - 1]
if n1 > n2:
die1_wins = die1_wins + 1
return die1_wins / num_rolls
print(roll_dice(blue_die, green_die))
0.338
print(roll_dice(green_die, orange_die))
0.3346
print(roll_dice(orange_die, yellow_die))
0.3358
print(roll_dice(yellow_die, blue_die))
0.3356