diff --git a/app/main.py b/app/main.py index fa56336e..8836cdb2 100644 --- a/app/main.py +++ b/app/main.py @@ -1 +1,36 @@ -# write your code here +import random +import matplotlib.pyplot as plt + + +def flip_coin_ten_times() -> int: + coin = ["head", "tail"] + head_counter = 0 + for i in range(10): + if random.choice(coin) == "head": + head_counter += 1 + return head_counter + + +def flip_coin() -> dict: + flip_dict = {} + total_cases = 10000 + for i in range(total_cases): + head_number = flip_coin_ten_times() + if head_number not in flip_dict: + flip_dict[head_number] = 1 + else: + flip_dict[head_number] += 1 + + for key in flip_dict: + flip_dict[key] = round((flip_dict[key] / total_cases) * 100, 2) + sorted_flip_dict = dict(sorted(flip_dict.items())) + + return sorted_flip_dict + + +def draw_gaussian_distribution_graph() -> None: + plt.ylim(0, 100) + plt.xlim(0, 10) + plt.ylabel("Drop percentage %") + plt.xlabel("Heads count") + plt.title("Gaussian") diff --git a/requirements.txt b/requirements.txt index 3f202d6e..893067e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,4 @@ flake8-quotes==3.3.1 flake8-variables-names==0.0.5 pep8-naming==0.13.2 pytest==7.1.3 +matplotlib