duplicate the level if a single level was passed

pydata · Mar 6, 2021 · 9d1c3cb · 9d1c3cb
1 parent 37522e9
commit 9d1c3cb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/xarray/plot/utils.py b/xarray/plot/utils.py
@@ -60,6 +60,11 @@ def _build_discrete_cmap(cmap, levels, extend, filled):
     """
     import matplotlib as mpl
 
+    if isinstance(levels, (int, float)):
+        levels = [levels, levels]
+    elif len(levels) == 1:
+        levels = [levels[0], levels[0]]
+
     if not filled:
         # non-filled contour plots
         extend = "max"