prob_calculator.py

import copy
import random
# Consider using the modules imported above.

class Hat():
    def __init__(self, **kwargs):
        self.contents = []
        for key, value in kwargs.items():
            for i in range(value):
                self.contents.append(key)
                
    def draw(self, draws_from_hat):
        if draws_from_hat > len(self.contents):
            return self.contents
        drawn_balls = []
        for i in range(draws_from_hat):
            d = random.randrange(len(self.contents))
            drawn_balls.append(self.contents.pop(d))
        return drawn_balls

        
def experiment(hat, expected_balls, num_balls_drawn, num_experiments):
    successes = 0
    for j in range(num_experiments):
        new_hat = copy.deepcopy(hat)
        experim = new_hat.draw(num_balls_drawn)
        
        balls_drawn = []
        for key, value in expected_balls.items():
            for i in range(value):
                balls_drawn.append(key)
    
        match_ball = 0
        
        for i in balls_drawn:
            if i in experim:
                experim.remove(i)
                match_ball += 1
    
        if match_ball == len(balls_drawn):
            successes += 1
    
    if successes == 0:
        return 0
    else:
        return successes / num_experiments