Merge pull request e2nIEE#590 from FlorianShepherd/develop

plot fix for sgen and load collection 🐑
AnBlav · Dec 24, 2019 · 5587aee · 5587aee
2 parents 288bcca + ac99a97
commit 5587aee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandapower/plotting/collections.py b/pandapower/plotting/collections.py
@@ -795,7 +795,7 @@ def create_load_collection(net, loads=None, size=1., infofunc=None, orientation=
     if loads is None:
         loads = net.load.index
     infos = [infofunc(i) for i in range(len(loads))] if infofunc is not None else []
-    node_coords = net.bus_geodata.loc[:, ["x", "y"]].values[net.load.loc[loads, "bus"].values]
+    node_coords = net.bus_geodata.loc[net.load.loc[loads, "bus"].values, ["x", "y"]].values
     load_pc, load_lc = _create_node_element_collection(
         node_coords, load_patches, size=size, infos=infos, orientation=orientation,
         picker=picker, **kwargs)
@@ -869,7 +869,7 @@ def create_sgen_collection(net, sgens=None, size=1., infofunc=None, orientation=
     if sgens is None:
         sgens = net.sgen.index
     infos = [infofunc(i) for i in range(len(sgens))] if infofunc is not None else []
-    node_coords = net.bus_geodata.loc[:, ["x", "y"]].values[net.sgen.loc[sgens, "bus"].values]
+    node_coords = net.bus_geodata.loc[net.sgen.loc[sgens, "bus"].values, ["x", "y"]].values
     sgen_pc, sgen_lc = _create_node_element_collection(
         node_coords, sgen_patches, size=size, infos=infos, orientation=orientation,
         picker=picker, **kwargs)