python code to create scatter plot