def survey_case(year):
# fname = "Agg_%s.%s" %(str(yr), "xls")
simulation = SurveySimulation()
simulation.set_config(year = year)
simulation.set_param()
# Ignore this
# inflator = get_loyer_inflator(year)
# simulation.inflate_survey({'loyer' : inflator})
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(year = 2006):
yr = str(year)
# fname = "Agg_%s.%s" %(str(yr), "xls")
simulation = SurveySimulation()
survey_filename = os.path.join(model.DATA_DIR, 'sources', 'test.h5')
simulation.set_config(year=yr, survey_filename=survey_filename)
simulation.set_param()
# Ignore this
# inflator = get_loyer_inflator(year)
# simulation.inflate_survey({'loyer' : inflator})
simulation.compute()
simul_out_df = simulation.output_table.table
simul_in_df = simulation.input_table.table
print simul_out_df.loc[:,['af', 'af_base', 'af_forf', 'af_majo', 'af_nbenf']].describe()
print 'input vars'
print simul_in_df.columns
print 'output vars'
print simul_out_df.columns
# check_inputs_enumcols(simulation)
# Compute aggregates
agg = Aggregates()
agg.set_simulation(simulation)
agg.compute()
df1 = agg.aggr_frame
print df1.columns
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()
return df1
def survey_case(year=2006):
yr = str(year)
# fname = "Agg_%s.%s" %(str(yr), "xls")
simulation = SurveySimulation()
survey_filename = os.path.join(model.DATA_DIR, 'sources', 'test.h5')
simulation.set_config(year=yr, survey_filename=survey_filename)
simulation.set_param()
# Ignore this
# inflator = get_loyer_inflator(year)
# simulation.inflate_survey({'loyer' : inflator})
simulation.compute()
simul_out_df = simulation.output_table.table
simul_in_df = simulation.input_table.table
print simul_out_df.loc[:,
['af', 'af_base', 'af_forf', 'af_majo', 'af_nbenf'
]].describe()
print 'input vars'
print simul_in_df.columns
print 'output vars'
print simul_out_df.columns
# check_inputs_enumcols(simulation)
# Compute aggregates
agg = Aggregates()
agg.set_simulation(simulation)
agg.compute()
df1 = agg.aggr_frame
print df1.columns
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()
return df1
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()
def get_agem_structure(simulation):
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df = pivot_table.get_table(entity = 'ind', by = "agem", vars = [])
return df
def get_structure(simulation, by_var):
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df = pivot_table.get_table(entity='ind', by=by_var, vars=[], champm=False)
return df
def get_age_structure(simulation):
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df = pivot_table.get_table(entity='ind', by="age", vars=[])
return df
def get_structure(simulation, by_var):
pivot_table = OpenfiscaPivotTable()
pivot_table.set_simulation(simulation)
df = pivot_table.get_table(entity = 'ind', by = by_var, vars = [], champm=False)
return df
