| import random |
| import streamlit as st |
| import altair as alt |
| import pandas as pd |
| import time |
|
|
| st.title("Monty Hall Problem Simulator") |
|
|
| def simulate(iterations, no_of_doors): |
| if no_of_doors < 3: |
| st.warning("Number of doors must be at least 3", icon="🙄") |
| return |
| switch_win = 0 |
| remain_win = 0 |
| data = [] |
|
|
| st.write(f"Simulation will loop for {iterations} iterations, with {no_of_doors} doors.") |
| progress_bar = st.progress(0) |
| chart_placeholder = st.empty() |
|
|
| for _iter in range(iterations): |
| doors = [0] * no_of_doors |
| car_door = random.randint(0, len(doors) - 1) |
| doors[car_door] = 1 |
| choice_door = random.randint(0, len(doors) - 1) |
|
|
| open_door = random.choice([i for i in range(len(doors)) if doors[i] == 0 and i != choice_door]) |
|
|
| if doors[choice_door] == 1: |
| remain_win += 1 |
| else: |
| switch_win += 1 |
|
|
| data.append({ |
| 'Iteration': _iter + 1, |
| 'Switch Wins %': (switch_win / (_iter + 1)) * 100, |
| 'Remain Wins %': (remain_win / (_iter + 1)) * 100 |
| }) |
|
|
| progress_bar.progress((_iter + 1) / iterations) |
| results_df = pd.DataFrame(data) |
| line_chart = alt.Chart(results_df).mark_line().encode( |
| x='Iteration:Q', |
| y='value:Q', |
| color='variable:N', |
| tooltip=['Iteration:Q', 'value:Q', 'variable:N'] |
| ).transform_fold( |
| ['Switch Wins %', 'Remain Wins %'], |
| as_=['variable', 'value'] |
| ).properties( |
| title='Winning Outcomes Over Iterations' |
| ) |
|
|
| chart_placeholder.altair_chart(line_chart, use_container_width=True) |
|
|
| time.sleep(0.1) |
|
|
| return pd.DataFrame(data) |
|
|
| with st.form("Input"): |
| no_of_doors = st.number_input("Enter number of doors", value=3) |
| iterations = st.number_input("Enter number of iterations", value=100) |
| state = st.form_submit_button("Start", type="primary") |
|
|
| if state: |
| if no_of_doors < 3: |
| st.warning("No of doors must be at least 3", icon="🙄") |
| else: |
| results_df = simulate(iterations, no_of_doors) |
| st.write(f"Winning scenarios for switching doors: {round(results_df['Switch Wins %'].iloc[-1]*iterations/100)}") |
| st.write(f"Winning scenarios for sticking to the same door: {round(results_df['Remain Wins %'].iloc[-1]*iterations/100)}") |
| st.write(f"Winning percentage for switching doors: {results_df['Switch Wins %'].iloc[-1]}%") |
| st.write(f"Winning percentage for sticking to the same door: {results_df['Remain Wins %'].iloc[-1]}%") |
