如何在matplotlib 1.4中使用viridis

Question

我想使用色彩映射表"viridis“(http://bids.github.io/colormap/)，而且我还不会更新到开发版本1.5。因此，我已经从https://github.com/BIDS/colormap下载了colormaps.py。不幸的是，我不能让它工作。这就是我要做的：

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

import colormaps as cmaps

img=mpimg.imread('stinkbug.png')
lum_img = np.flipud(img[:,:,0])

plt.set_cmap(cmaps.viridis)
imgplot = plt.pcolormesh(lum_img)

这给了我一个ValueError，回溯以，

ValueError:无法识别Colormap viridis。可能的值有:光谱、夏季、凉爽、...

(然后是最初安装的色彩映射表的完整列表。)

对如何解决这个问题有什么想法吗？

Answer 1

要使用set_cmap将viridis设置为色彩贴图，必须先注册它：

import colormaps as cmaps
plt.register_cmap(name='viridis', cmap=cmaps.viridis)
plt.set_cmap(cmaps.viridis)

img=mpimg.imread('stinkbug.png')
lum_img = np.flipud(img[:,:,0])
imgplot = plt.pcolormesh(lum_img)

Answer 2

您可以直接在pcolormesh调用中设置cmap，而不是使用需要matplotlib.colors.Colormap实例的set_cmap

(cmaps.viridis是一个matplotlib.colors.ListedColormap)

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

import colormaps as cmaps

img=mpimg.imread('stinkbug.png')
lum_img = np.flipud(img[:,:,0])

imgplot = plt.pcolormesh(lum_img, cmap=cmaps.viridis)

Answer 3

我所做的就是复制

_viridis_data = [[0.267004, 0.004874, 0.329415],
                 [0.268510, 0.009605, 0.335427],
                 [0.269944, 0.014625, 0.341379],
                 :
                 [0.983868, 0.904867, 0.136897],
                 [0.993248, 0.906157, 0.143936]]

来自https://github.com/BIDS/colormap/blob/master/colormaps.py

并添加：

from matplotlib.colors import ListedColormap

viridis = ListedColormap(_viridis_data, name='viridis')

plt.register_cmap(name='viridis', cmap=viridis)
plt.set_cmap(viridis)