| 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]}%") |