aideencaulfield/Streamlit-Data-Synthesis-Example

1import streamlit as st2import pandas as pd 3def generate_hospital_data():4    # Generate hospital data5    hospitals = {6        "city": ["New York", "Los Angeles", "Chicago", "Houston", "Phoenix"],7        "state": ["NY", "CA", "IL", "TX", "AZ"],8        "bed_count": [1200, 1500, 1100, 1300, 1400],9    }10    df = pd.DataFrame(hospitals)11    return df 12def generate_state_data():13    # Generate state data14    states = {15        "state": ["NY", "CA", "IL", "TX", "AZ"],16        "population": [20000000, 40000000, 13000000, 29000000, 7000000],17        "square_miles": [54556, 163696, 57914, 268596, 113990],18    }19    df = pd.DataFrame(states)20    return df 21def merge_datasets(hospitals_df, states_df):22    # Merge hospital and state data23    merged_df = pd.merge(hospitals_df, states_df, on="state")24    return merged_df 25def calculate_beds_per_capita(merged_df):26    # Calculate beds per capita27    merged_df["beds_per_capita"] = merged_df["bed_count"] / merged_df["population"]28    return merged_df 29def main():30    # Generate data31    hospitals_df = generate_hospital_data()32    states_df = generate_state_data()     # Merge datasets33    merged_df = merge_datasets(hospitals_df, states_df)     # Calculate beds per capita34    merged_df = calculate_beds_per_capita(merged_df)     # Show merged and calculated data35    st.write(merged_df) 36if __name__ == "__main__":37    main()38 39