Python scatter plot and