Label data when doing a scatter plot in python,plt.legend()