Matplotlib scatterplot; colour as a function of a third variable,plt.show()