Matplotlib make a scatter plot figure