Allow viridis to be used by name

2026-07-05 19:06:06 +08:00 · 2015-07-16 07:58:14 -05:00
parent 3d963649fc
commit eaaea690f2
1 changed files with 2 additions and 0 deletions
@@ -140,6 +140,8 @@ def imshow(im, *args, **kwargs):
    ax_im : `matplotlib.pyplot.AxesImage`
        The `AxesImage` object returned by `plt.imshow`.
    """
+    if kwargs.get('cmap', None) == 'viridis':
+        kwargs['cmap'] = viridis
    lo, hi, cmap = _get_display_range(im)
    kwargs.setdefault('interpolation', 'nearest')
    kwargs.setdefault('cmap', cmap)