[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-03-06 19:43:30 +00:00
parent a8682db4fc
commit d2968230bb

View File

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