Add legend to scatter plot (PCA)

by

I recently proposed an easy way to add a legend to a scatter, see GitHub PR. This is still being discussed.

In the meantime you need to manually create your legend from the unique labels in y. For each of them you’d create a Line2D object with the same marker as is used in the scatter plot and supply them as argument to plt.legend.

scatter = plt.scatter(xs * scalex,ys * scaley, c = y)
labels = np.unique(y)
handles = [plt.Line2D([],[],marker="o", ls="", 
                      color=scatter.cmap(scatter.norm(yi))) for yi in labels]
plt.legend(handles, labels)

enter image description here

More Related Contents:

  1. How to put the legend outside the plot in Matplotlib
  2. Secondary axis with twinx(): how to add to legend?
  3. Moving matplotlib legend outside of the axis makes it cutoff by the figure box
  4. How to manually create a legend
  5. How to put the legend outside the plot
  6. creating over 20 unique legend colors using matplotlib
  7. How to edit a seaborn legend title and labels for figure-level functions
  8. Matplotlib scatter plot with legend
  9. Stop matplotlib repeating labels in legend
  10. How to change legend size with matplotlib.pyplot
  11. matplotlib: colorbars and its text labels
  12. Subheadings for categories within matplotlib custom legend
  13. How to change the text color of font in legend?
  14. Styling part of label in legend in matplotlib
  15. How to change legend fontsize with matplotlib.pyplot
  16. How to add a string as the artist in matplotlib legend?
  17. How is order of items in matplotlib legend determined?
  18. Change main plot legend label text
  19. Is it possible to add a string as a legend item in matplotlib
  20. using mpatches.Patch for a custom legend
  21. matplotlib: 2 different legends on same graph
  22. Remove line through marker in matplotlib legend
  23. Matplotlib savefig with a legend outside the plot
  24. Combine two Pyplot patches for legend
  25. Adding a matplotlib legend
  26. How to plot a scatter plot with a legend label for each class
  27. Scatter plot with legend for each color in c
  28. Legend overlaps with the pie chart
  29. Is it possible to add a string as a legend item
  30. Adding legend based on existing color series

Leave a Comment