Simulation of Monty Hall Problem

monty_hall_problem_3.py

# Perform a empirical simulation
data_switch = monty_hall(
    num_iter = 1000,
    switch = True
)

# Create a data rame
df_switch = pd.DataFrame(
    data = data_switch
)

# Create a column of probability
df_switch['win_rate'] = (df_switch['win'] / df_switch['iter']).apply(
    lambda x: round(x, 3)
)
df_switch['lose_rate'] = (1 - df_switch['win_rate']).apply(
    lambda x: round(x, 3)
)

# Show data frame
df_switch.head()