fix: set dropna to false to keep matrix dimensions (#605)

Breakthrough-Energy · Mar 16, 2022 · b2c1dd6 · b2c1dd6
1 parent f27ebcf
commit b2c1dd6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/powersimdata/input/input_data.py b/powersimdata/input/input_data.py
@@ -168,7 +168,11 @@ def distribute_demand_from_zones_to_buses(zone_demand, bus):
         [pd.Series(bus["Pd"] / bus_zone_pd, name="zone_share"), bus["zone_id"]], axis=1
     )
     zone_bus_shares = bus_zone_share.pivot_table(
-        index="bus_id", columns="zone_id", values="zone_share", fill_value=0
+        index="bus_id",
+        columns="zone_id",
+        values="zone_share",
+        dropna=False,
+        fill_value=0,
     )
     bus_demand = zone_demand.dot(zone_bus_shares.T)