def test():
country = "france"
for year in range(2006,2010):
yr = str(year)
simu = SurveySimulation()
simu.set_config(year = yr, country = country)
simu.set_param()
agg = Aggregates()
agg.set_simulation(simu)
for col in agg.varlist:
print col
def test():
country = "france"
for year in range(2006, 2010):
yr = str(year)
simu = SurveySimulation()
simu.set_config(year=yr, country=country)
simu.set_param()
agg = Aggregates()
agg.set_simulation(simu)
for col in agg.varlist:
print col
def build_erf_aggregates():
"""
Fetch the relevant aggregates from erf data
"""
# Uses rpy2.
# On MS Windows, The environment variable R_HOME and R_USER should be set
import pandas.rpy.common as com
import rpy2.rpy_classic as rpy
rpy.set_default_mode(rpy.NO_CONVERSION)
country = 'france'
for year in range(2006, 2008):
menageXX = "menage" + str(year)[2:]
menageRdata = menageXX + ".Rdata"
filename = os.path.join(os.path.dirname(DATA_DIR), 'R', 'erf',
str(year), menageRdata)
yr = str(year)
simu = SurveySimulation()
simu.set_config(year=yr, country=country)
simu.set_param()
agg = Aggregates()
agg.set_simulation(simu)
# print agg.varlist
rpy.r.load(filename)
menage = com.load_data(menageXX)
cols = []
print year
for col in agg.varlist:
#print col
erf_var = "m_" + col + "m"
if erf_var in menage.columns:
cols += [erf_var]
df = menage[cols]
wprm = menage["wprm"]
for col in df.columns:
tot = (df[col] * wprm).sum() / 1e9
print col, tot
def build_erf_aggregates():
"""
Fetch the relevant aggregates from erf data
"""
# Uses rpy2.
# On MS Windows, The environment variable R_HOME and R_USER should be set
import pandas.rpy.common as com
import rpy2.rpy_classic as rpy
rpy.set_default_mode(rpy.NO_CONVERSION)
country = 'france'
for year in range(2006,2008):
menageXX = "menage" + str(year)[2:]
menageRdata = menageXX + ".Rdata"
filename = os.path.join(os.path.dirname(DATA_DIR),'R','erf', str(year), menageRdata)
yr = str(year)
simu = SurveySimulation()
simu.set_config(year = yr, country = country)
simu.set_param()
agg = Aggregates()
agg.set_simulation(simu)
# print agg.varlist
rpy.r.load(filename)
menage = com.load_data(menageXX)
cols = []
print year
for col in agg.varlist:
#print col
erf_var = "m_" + col + "m"
if erf_var in menage.columns:
cols += [erf_var]
df = menage[cols]
wprm = menage["wprm"]
for col in df.columns:
tot = (df[col]*wprm).sum()/1e9
print col, tot
