color coded scatter plot python