Matplotlib colorbar for one subplot. pyplt as plt ax = plt.