Add equal plot scales

Author: gabrielaznar

climada/util/plot.py

@@ -62,6 +62,13 @@ def geo_bin_from_array(array_sub, geo_coord, var_name, title, pop_name=True,
 
     if 'cmap' not in kwargs:
         kwargs['cmap'] = 'Wistia'
+    if 'vmin' not in kwargs:
+        if ignore_zero:
+            kwargs['vmin'] = np.min(array_sub[array_sub > 0])
+        else:
+            kwargs['vmin'] = np.min(array_sub)
+    if 'vmax' not in kwargs:
+        kwargs['vmax'] = np.max(array_sub)
 
     # Generate each subplot
     fig, axis_sub = make_map(num_im)
@@ -70,7 +77,7 @@ def geo_bin_from_array(array_sub, geo_coord, var_name, title, pop_name=True,
         if coord.shape[0] != array_im.size:
             raise ValueError("Size mismatch in input array: %s != %s." % \
                              (coord.shape[0], array_im.size))
-        if ignore_zero:
+        if ignore_zero and 0 in array_im:
             pos_vals = array_im > 0
             array_im = array_im[pos_vals]
             coord = coord[pos_vals, :]