bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Reference to a multiindex level by number instead of name

Browse Source
This commit is contained in:
ekatef 2023-10-11 12:29:38 +03:00
parent 7ad60cb7f9
commit fcef194566
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/build_retro_cost.py
View File

@ -990,19 +990,22 @@ def sample_dE_costs_area(
sec_w = area_tot.div(area_tot.groupby(level=0).transform('sum')) sec_w = area_tot.div(area_tot.groupby(level=0).transform('sum'))
# get the total cost-energy-savings weight by sector area # get the total cost-energy-savings weight by sector area
tot = ( tot = (
cost_dE.mul(sec_w, axis=0) # sec_w has columns "estimated" and "value"
.groupby(level="country_code") cost_dE.mul(sec_w.value, axis=0)
# for some reasons names of the levels were lost somewhere
#.groupby(level="country_code")
.groupby(level=0)
.sum() .sum()
.set_index( .set_index(
pd.MultiIndex.from_product( pd.MultiIndex.from_product(
[cost_dE.index.unique(level="country_code"), ["tot"]] [cost_dE.index.unique(level=0), ["tot"]]
) )
) )
) )
cost_dE = pd.concat([cost_dE, tot]).unstack().stack() cost_dE = pd.concat([cost_dE, tot]).unstack().stack()
summed_area = pd.DataFrame(area_tot.groupby("country").sum()).set_index( summed_area = pd.DataFrame(area_tot.groupby(level=0).sum()).set_index(
pd.MultiIndex.from_product([area_tot.index.unique(level="country"), ["tot"]]) pd.MultiIndex.from_product([area_tot.index.unique(level=0), ["tot"]])
) )
area_tot = pd.concat([area_tot, summed_area]).unstack().stack() area_tot = pd.concat([area_tot, summed_area]).unstack().stack()