Création d'une légende dans Colormap Matplotlib

Question

J'utilise imshow() dans matplotlib comme ceci:

import numpy as np 
import matplotlib.pyplot as plt 
mat = '''SOME MATRIX''' 
plt.imshow(mat, origin="lower", cmap='gray', interpolation='nearest') 
plt.show() 

Comment puis-je ajouter une légende indiquant la valeur numérique pour les différentes nuances de gris. Malheureusement, mon googler n'a pas découvert une réponse :(

Merci à l'avance pour l'aide

Vince

Answer 1

Simple, il suffit plt.colorbar():

import numpy as np 
import matplotlib.pyplot as plt 
mat = np.random.random((10,10)) 
plt.imshow(mat, origin="lower", cmap='gray', interpolation='nearest') 
plt.colorbar() 
plt.show() 

Answer 2

Comme d'habitude, je le figure juste après que je le demande,.). Pour la postérité, voici mon coup de poing:

m = np.zeros((1,20)) 
for i in range(20): 
    m[0,i] = (i*5)/100.0 
print m 
plt.imshow(m, cmap='gray', aspect=2) 
plt.yticks(np.arange(0)) 
plt.xticks(np.arange(0,25,5), [0,25,50,75,100]) 
plt.show() 

Je suis sûr qu'il existe une solution plus élégante.

Vince

Answer 3

Il y a un colorbar() function dans builtin pyplot.