Come utilizzare viridis in matplotlib 1.4

Question

Voglio usare la mappa dei colori "viridis" (http://bids.github.io/colormap/) e non aggiornerò ancora la versione di sviluppo 1.5. Così, ho scaricato colormaps.py da https://github.com/BIDS/colormap. Sfortunatamente, non sono in grado di farlo funzionare. Questo è quello che faccio:

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) 

Questo mi dà una ValueError, il traceback termina con,

ValueError: Colormap viridis is not recognized. Possible values are: Spectral, summer, coolwarm, ...

(E poi la lista completa delle mappe di colore installati originariamente.)

Ogni pensiero su come risolvere questo problema?

Answer 1

Per impostare viridis come mappa di colori usando set_cmap , devi prima registrarlo:

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

Piuttosto che usare set_cmap, che richiede un'istanza matplotlib.colors.Colormap, è possibile impostare la cmap direttamente nella pcolormesh chiamata

(cmaps.viridis è un 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

Quello che ho fatto è quello di copiare il

_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]] 

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

e aggiungere:

from matplotlib.colors import ListedColormap 

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

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

Answer 4

Scarica il colormaps.py da here, quindi:

import os,sys 
scriptpath = "/Your downloading path/colormap-master/" 
sys.path.append(os.path.abspath(scriptpath)) 
import colormaps as cmaps 

Done !