matplotlib scatter plot colour as function of third variable,plt.show()