Add a legend in a 3D scatterplot with scatter() in Matplotlib

Add a legend in a 3D scatterplot with scatter() in Matplotlib