From 39e853e2b363427a52b3dc056c335a1365ebd1a7 Mon Sep 17 00:00:00 2001 From: lisazeyen Date: Fri, 19 Jul 2024 15:34:44 +0200 Subject: [PATCH] make sure not to divide by zero --- scripts/build_energy_totals.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/build_energy_totals.py b/scripts/build_energy_totals.py index 70ed4bec..4e353c11 100644 --- a/scripts/build_energy_totals.py +++ b/scripts/build_energy_totals.py @@ -847,7 +847,7 @@ def build_district_heat_share(countries: List[str], idees: pd.DataFrame) -> pd.S ) total_heat = idees[["thermal uses residential", "thermal uses services"]].sum( axis=1 - ) + ).replace(0, np.nan) district_heat_share = district_heat / total_heat @@ -866,6 +866,8 @@ def build_district_heat_share(countries: List[str], idees: pd.DataFrame) -> pd.S district_heat_share = pd.concat( [district_heat_share, dh_share.reindex(new_index, level=0)], axis=1 ).min(axis=1) + + district_heat_share = district_heat_share.reindex(countries, level=0) district_heat_share.name = "district heat share"