Guest User

Untitled

a guest
Feb 25th, 2018
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.80 KB | None | 0 0
  1. import matplotlib.pyplot as plt
  2. import seaborn as sns
  3.  
  4. from matplotlib.lines import Line2D
  5.  
  6. my_palette = sns.color_palette("cubehelix", 3)
  7. sns.set_palette(my_palette)
  8.  
  9. def legend_circles(labels, palette, loc=1, markersize=10, marker='o', padding=0):
  10. "Make a legend where the color is indicated by a circle."
  11. legend_markers = [Line2D(range(1), range(1),
  12. linewidth=0, # Invisible line
  13. marker=marker,
  14. markersize=markersize,
  15. markerfacecolor=palette[i]) for i in range(len(labels))]
  16. return plt.legend(legend_markers, labels, numpoints=1, loc=loc, handletextpad=padding)
  17.  
  18. plt.plot([1,2,3])
  19. plt.plot([0,1,3])
  20. plt.plot([3,2,1])
  21.  
  22. legend = legend_circles(['a','b','c'], my_palette,loc=0)
  23. plt.show()
Add Comment
Please, Sign In to add comment