Fix peak load: calculate per scenario

openego · Nov 9, 2022 · 5da3c55 · 5da3c55
1 parent 7154fd5
commit 5da3c55
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/egon/data/datasets/electricity_demand_timeseries/cts_buildings.py b/src/egon/data/datasets/electricity_demand_timeseries/cts_buildings.py
@@ -1564,9 +1564,9 @@ def assign_voltage_level_to_buildings():
             cells_query.session.bind,
         )
 
-    df_peak_load_buildings = df_peak_loads.groupby("building_id")[
-        "peak_load_in_w"
-    ].sum()
+    df_peak_load_buildings = df_peak_loads.groupby(
+        ["building_id", "scenario"]
+    )["peak_load_in_w"].sum()
     df_peak_load_buildings = df_peak_load_buildings.to_frame()
     df_peak_load_buildings.loc[:, "voltage_level"] = 0
 
@@ -1594,7 +1594,7 @@ def assign_voltage_level_to_buildings():
         left=df_peak_loads.drop(columns="voltage_level"),
         right=df_peak_load_buildings["voltage_level"],
         how="left",
-        left_on="building_id",
+        left_on=["building_id", "scenario"],
         right_index=True,
     )