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()