Creación de una leyenda de mapa de color en Matplotlib

Question

estoy usando imshow() en matplotlib así:

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

¿Cómo agrego una leyenda que muestra el valor numérico de los diferentes tonos de gris. Por desgracia, mi googlear no ha puesto al descubierto una respuesta :(

Gracias de antemano por la ayuda

Vince

Answer 1

simple, simplemente 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

Como de costumbre, lo averigüe justo después de lo pido;.). Para la posteridad, aquí está mi puñalada en ella:

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() 

Estoy seguro de que existe una solución más elegante.

Vince

Answer 3

Hay una orden interna colorbar() function en pyplot.