pyplot: define customised colormap for image plot which assigns a color to a certain value

Question

What is the easiest way to display an array with only very few different values as an image plot, e.g. with imshow from matplotlib.pyplot, assignin