[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
a8682db4fc
commit
d2968230bb
@ -859,15 +859,17 @@ def rescale_idees_from_eurostat(
|
|||||||
slicer_target = idx[country, 2016:2021, :, :]
|
slicer_target = idx[country, 2016:2021, :, :]
|
||||||
slicer_source = idx[country, 2015, :, :]
|
slicer_source = idx[country, 2015, :, :]
|
||||||
for sector, mapping in mappings.items():
|
for sector, mapping in mappings.items():
|
||||||
sector_ratio = ratio.loc[(country, slice(None), slice(None), sector)].droplevel("lvl2")
|
sector_ratio = ratio.loc[
|
||||||
|
(country, slice(None), slice(None), sector)
|
||||||
|
].droplevel("lvl2")
|
||||||
|
|
||||||
energy.loc[slicer_target, mapping["total"]] = cartesian(
|
energy.loc[slicer_target, mapping["total"]] = cartesian(
|
||||||
sector_ratio.loc[2016:2021, "total"],
|
sector_ratio.loc[2016:2021, "total"],
|
||||||
energy.loc[slicer_source, mapping["total"]].squeeze()
|
energy.loc[slicer_source, mapping["total"]].squeeze(),
|
||||||
).values
|
).values
|
||||||
energy.loc[slicer_target, mapping["elec"]] = cartesian(
|
energy.loc[slicer_target, mapping["elec"]] = cartesian(
|
||||||
sector_ratio.loc[2016:2021, "ele"],
|
sector_ratio.loc[2016:2021, "ele"],
|
||||||
energy.loc[slicer_source, mapping["elec"]].squeeze()
|
energy.loc[slicer_source, mapping["elec"]].squeeze(),
|
||||||
).values
|
).values
|
||||||
|
|
||||||
level_drops = ["country", "lvl2", "lvl3"]
|
level_drops = ["country", "lvl2", "lvl3"]
|
||||||
@ -882,18 +884,15 @@ def rescale_idees_from_eurostat(
|
|||||||
nav = ratio.loc[slicer, "total"].droplevel(level_drops)
|
nav = ratio.loc[slicer, "total"].droplevel(level_drops)
|
||||||
|
|
||||||
energy.loc[slicer_target, avia_inter] = cartesian(
|
energy.loc[slicer_target, avia_inter] = cartesian(
|
||||||
avi_i.loc[2016:2021],
|
avi_i.loc[2016:2021], energy.loc[slicer_source, avia_inter].squeeze()
|
||||||
energy.loc[slicer_source, avia_inter].squeeze()
|
|
||||||
).values
|
).values
|
||||||
|
|
||||||
energy.loc[slicer_target, avia_domestic] = cartesian(
|
energy.loc[slicer_target, avia_domestic] = cartesian(
|
||||||
avi_d.loc[2016:2021],
|
avi_d.loc[2016:2021], energy.loc[slicer_source, avia_domestic].squeeze()
|
||||||
energy.loc[slicer_source, avia_domestic].squeeze()
|
|
||||||
).values
|
).values
|
||||||
|
|
||||||
energy.loc[slicer_target, navigation] = cartesian(
|
energy.loc[slicer_target, navigation] = cartesian(
|
||||||
nav.loc[2016:2021],
|
nav.loc[2016:2021], energy.loc[slicer_source, navigation]
|
||||||
energy.loc[slicer_source, navigation]
|
|
||||||
).values
|
).values
|
||||||
|
|
||||||
return energy
|
return energy
|
||||||
@ -925,9 +924,7 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
# Data from IDEES only exists from 2000-2015.
|
# Data from IDEES only exists from 2000-2015.
|
||||||
logger.info("Extrapolate IDEES data based on eurostat for years 2015-2021.")
|
logger.info("Extrapolate IDEES data based on eurostat for years 2015-2021.")
|
||||||
energy = rescale_idees_from_eurostat(
|
energy = rescale_idees_from_eurostat(idees_countries, energy, eurostat)
|
||||||
idees_countries, energy, eurostat
|
|
||||||
)
|
|
||||||
|
|
||||||
energy.to_csv(snakemake.output.energy_name)
|
energy.to_csv(snakemake.output.energy_name)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user