Remove title from colorbar (#260)

And add units to its label instead
pySTEPS · Jan 27, 2022 · 6752bad · 6752bad
1 parent 10540d2
commit 6752bad
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/pysteps/visualization/precipfields.py b/pysteps/visualization/precipfields.py
@@ -191,13 +191,11 @@ def plot_precip_field(
             cbar.ax.set_yticklabels(clevs_str)
 
         if ptype == "intensity":
-            cbar.ax.set_title(units, fontsize=10)
-            cbar.set_label("Precipitation intensity")
+            cbar.set_label(f"Precipitation intensity [{units}]")
         elif ptype == "depth":
-            cbar.ax.set_title(units, fontsize=10)
-            cbar.set_label("Precipitation depth")
+            cbar.set_label(f"Precipitation depth [{units}]")
         else:
-            cbar.set_label("P(R > %.1f %s)" % (probthr, units))
+            cbar.set_label(f"P(R > {probthr:.1f} {units})")
 
     if geodata is None or axis == "off":
         ax.xaxis.set_ticks([])