Merge pull request #235 from pypsa-meets-earth/fix_bug_demandworkflow

fix minor bug
pypsa-meets-earth · Sep 14, 2023 · 5537672 · 5537672
2 parents 97c2f2f + c5a08bb
commit 5537672
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/scripts/prepare_energy_totals.py b/scripts/prepare_energy_totals.py
@@ -81,7 +81,11 @@ def calculate_end_values(df):
         options["shipping_hydrogen_share"], demand_sc + "_" + str(investment_year)
     )
 
-    energy_totals = base_energy_totals * efficiency_gains * growth_factors
+    energy_totals = (
+        base_energy_totals
+        * efficiency_gains.loc[countries]
+        * growth_factors.loc[countries]
+    )
 
     # Residential
     efficiency_heat_oil_to_elec = snakemake.config["sector"][
@@ -270,4 +274,4 @@ def calculate_end_values(df):
 
     energy_totals = energy_totals.dropna(axis=1, how="all")
 
-    energy_totals.to_csv(snakemake.output.energy_totals)
+    energy_totals.fillna(0).to_csv(snakemake.output.energy_totals)