43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import copy
|
|
import random
|
|
|
|
|
|
# Consider using the modules imported above.
|
|
|
|
class Hat:
|
|
|
|
def __init__(self, **kwargs):
|
|
self.contents = []
|
|
for arg in kwargs:
|
|
for i in range(0, kwargs[arg]):
|
|
self.contents.append(arg)
|
|
|
|
def draw(self, amount):
|
|
temp = copy.copy(self.contents)
|
|
drawn = []
|
|
if amount >= len(temp):
|
|
return temp
|
|
for i in range(0, amount):
|
|
rand = random.choice(temp)
|
|
drawn.append(rand)
|
|
temp.pop(temp.index(rand))
|
|
self.contents = temp
|
|
return drawn
|
|
|
|
|
|
def experiment(hat, expected_balls, num_balls_drawn, num_experiments):
|
|
success = 0
|
|
for i in range(0, num_experiments):
|
|
print("experiment:", i)
|
|
test = copy.deepcopy(hat)
|
|
balls = test.draw(num_balls_drawn)
|
|
counts = {}
|
|
match = True
|
|
for ball in balls:
|
|
counts[ball] = counts.get(ball, 0) + 1
|
|
for color in expected_balls:
|
|
if counts.get(color, 0) < expected_balls[color]:
|
|
match = False
|
|
if match:
|
|
success = success + 1
|
|
return success / num_experiments
|