generated from mwc/lab_dice
77 lines
1.9 KiB
Python
77 lines
1.9 KiB
Python
from pickle import FALSE
|
|
from die import Die
|
|
from tqdm import tqdm
|
|
|
|
class FiveDice:
|
|
def __init__(self):
|
|
self.dice = [Die() for number in range(5)]
|
|
|
|
def roll(self):
|
|
for die in self.dice:
|
|
die.roll()
|
|
return self.faces()
|
|
|
|
def faces(self):
|
|
return [die.face for die in self.dice]
|
|
|
|
def all_ones(self):
|
|
for face in self.faces():
|
|
if face != 1:
|
|
return False
|
|
return True
|
|
def is_three_of_a_kind(self):
|
|
facesList = self.faces()
|
|
facesList.sort()
|
|
if facesList[0]==facesList[2]:
|
|
return True
|
|
elif facesList[1]==facesList[3]:
|
|
return True
|
|
elif facesList[3]==facesList[5]:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def is_four_of_a_kind(self):
|
|
facesList = self.faces()
|
|
facesList.sort()
|
|
if facesList[0]==facesList[3]:
|
|
return True
|
|
elif facesList[1]==facesList[4]:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def different_three_of_a_kind(self):
|
|
facesList = self.faces()
|
|
for i in range(0, 2, 1):
|
|
maxCount = 0
|
|
for j in range(i, 4, 1):
|
|
if facesList[i] == facesList[j]:
|
|
maxCount += 1
|
|
if maxCount >= 3:
|
|
return True
|
|
return False
|
|
|
|
def different_four_of_a_kind(self):
|
|
facesList = self.faces()
|
|
for i in range(0, 1, 1):
|
|
maxCount = 0
|
|
for j in range(i, 4, 1):
|
|
if facesList[i] == facesList[j]:
|
|
maxCount += 1
|
|
if maxCount >= 4:
|
|
return True
|
|
return False
|
|
|
|
dice = FiveDice()
|
|
successes = 0
|
|
trials = 1000000
|
|
for trial in tqdm(range(trials)):
|
|
dice.roll()
|
|
if dice.all_ones():
|
|
successes += 1
|
|
|
|
print(successes/trials)
|
|
|