How to add legend to imshow() in matplotlib

Question

I am using matplotlib

In plot() or bar(), we can easily put legend, if we add labels to them. but what if it is a contou

Answer 1

I guess you have to fake your legend, since it requires a line for creating the legend.

You can do something like this:

import pylab as pl
mycmap = pl.cm.jet # for example
for entry in pl.unique(raw_data):
    mycolor = mycmap(entry*255/(max(raw_data) - min(raw_data)))
    pl.plot(0, 0, "-", c=mycolor, label=mynames[entry])

pl.imshow(raw_data)
pl.legend()

Of cause this is not very satisfying yet. But maybe you can build something on that.

[edit: added missing parenthesis]