use bus voltage results in create_bus_collection if z is None

kyri-petrou · Jun 18, 2020 · df77d11 · df77d11
1 parent af04313
commit df77d11
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pandapower/plotting/collections.py b/pandapower/plotting/collections.py
@@ -381,6 +381,8 @@ def create_bus_collection(net, buses=None, size=5, patch_type="circle", color=No
     pc = _create_node_collection(buses, coords, size, patch_type, color, picker, infos, **kwargs)
 
     if cmap is not None:
+        if z is None:
+            z = net.res_bus.vm_pu.loc[buses]
         add_cmap_to_collection(pc, cmap, norm, z, cbar_title)
 
     return pc