matplotlib: how to change data points color based on some variable,plt.show()