Extend _register_cmap_clip()

mgeier · mgeier · commit 14d6dace91f4 · 2019-03-19T11:50:10.000+01:00
diff --git a/sfs/plot2d.py b/sfs/plot2d.py
@@ -13,8 +13,11 @@
 def _register_cmap_clip(name, original_cmap, alpha):
     """Create a color map with "over" and "under" values."""
     from matplotlib.colors import LinearSegmentedColormap
-    cdict = plt.cm.datad[original_cmap]
-    cmap = LinearSegmentedColormap(name, cdict)
+    cdata = plt.cm.datad[original_cmap]
+    if isinstance(cdata, dict):
+        cmap = LinearSegmentedColormap(name, cdata)
+    else:
+        cmap = LinearSegmentedColormap.from_list(name, cdata)
     cmap.set_over([alpha * c + 1 - alpha for c in cmap(1.0)[:3]])
     cmap.set_under([alpha * c + 1 - alpha for c in cmap(0.0)[:3]])
     plt.cm.register_cmap(cmap=cmap)
