python - 如何在 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 is not recognized. Possible values are: Spectral, summer, coolwarm, ...

(然后是最初安装的颜色图的完整列表。)

关于如何解决此问题的任何想法？

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)