Defining a discrete colormap for imshow in matplotlib,plt.show()