with - plot legend python

Creando una leyenda de mapa de colores 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 para los diferentes tonos de gris? Lamentablemente, mi google no ha descubierto una respuesta :(

Gracias por adelantado por la ayuda.

Vince

Answer 1

Como de costumbre, lo resuelvo justo después de que lo pregunto;). Para la posteridad, aquí está mi puñalada:

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 2

Hay una función integrada colorbar () en pyplot.

Answer 3

Simple, solo 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()