application_vnd.jupyter.stderr_output_1.png
import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

case_c = pd.read_csv('/kaggle/input/group-c-data/analysis_data_test.csv')

# reshape dataframe into pivot table
d = case_c.pivot(index = 'x_value', columns = 'n_value', values = 'avg_peak_bed_usage')
# set visualization features like title, axis names, and legend name
d.columns.name = 'Initial Infections'
plot1 = d.plot(title = 'Policy Strictness v. Hospital Bed Demand')
plot1.set(xlabel = 'Maximum Allowed Group Size', ylabel = 'Hospital Bed Demand')

f = case_c.pivot(index='x_value', columns='n_value', values='success_rate')
f.columns.name = 'Initial Infections'
plot2 = f.plot(title='Policy Strictness v. Simulation Success Rate')
plot2.set(xlabel='Maximum Allowed Group Size', ylabel='Success Rate')