Создание легенды Colormap в Matplotlib

Я использую imshow() в matplotlib так:

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

Как добавить легенду, показывающую числовое значение для разных оттенков серого. К сожалению, мой поисковик не обнаружил ответа: (

Заранее спасибо за помощь.

Винс

Ответ 1

Просто, просто 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()

Ответ 2

Там есть встроенная функция colorbar() в pyplot.

Ответ 3

Как обычно, я выясняю это сразу после того, как я его прошу;). Для потомков, здесь мой удар по нему:

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

Я уверен, что существует более элегантное решение.

Винс