def survey_case_3_tables():
year = 2006
yr = str(year)
simulation = SurveySimulation()
survey_input = HDFStore(survey3_test)
# convert_to_3_tables(year=year, survey_file=survey_file, output_file=survey3_file)
df_men = survey_input['survey_2006/men']
df_foy = survey_input['survey_2006/foy']
df_fam = survey_input['survey_2006/fam']
df_fam['alr'] = 0;
survey_input['survey_2006/fam'] = df_fam
simulation.num_table = 3
simulation.set_config(year = yr, survey_filename=survey3_test)
simulation.set_param()
simulation.compute()
# Compute aggregates
agg = Aggregates()
agg.set_simulation(simulation)
agg.compute()
df1 = agg.aggr_frame
print df1.to_string()
# # Saving aggregates
# if writer is None:
# writer = ExcelWriter(str(fname)
# agg.aggr_frame.to_excel(writer, yr, index= False, header= True)
# Displaying a pivot table
from openfisca_qt.plugins.survey.distribution import OpenfiscaPivotTable
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df2 = pivot_table.get_table(by ='so', vars=['nivvie'])
print df2.to_string()
def survey_case_3_tables():
year = 2006
yr = str(year)
simulation = SurveySimulation()
survey_input = HDFStore(survey3_test)
# convert_to_3_tables(year=year, survey_file=survey_file, output_file=survey3_file)
df_men = survey_input['survey_2006/men']
df_foy = survey_input['survey_2006/foy']
df_fam = survey_input['survey_2006/fam']
df_fam['alr'] = 0
survey_input['survey_2006/fam'] = df_fam
simulation.num_table = 3
simulation.set_config(year=yr, survey_filename=survey3_test)
simulation.set_param()
simulation.compute()
# Compute aggregates
agg = Aggregates()
agg.set_simulation(simulation)
agg.compute()
df1 = agg.aggr_frame
print df1.to_string()
# # Saving aggregates
# if writer is None:
# writer = ExcelWriter(str(fname)
# agg.aggr_frame.to_excel(writer, yr, index= False, header= True)
# Displaying a pivot table
from openfisca_qt.plugins.survey.distribution import OpenfiscaPivotTable
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df2 = pivot_table.get_table(by='so', vars=['nivvie'])
print df2.to_string()
