Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Matplotlib : making a colored markers legend from scratch

Tags:

In Matplotlib, i'm trying to make a legend with colored "markers" like this one :

Markers legend

this one has been made using the scatter function, but is not adapted to my plot. I'd like to produce a legend "from scratch", without associated data. The color is important, and therefore should be an attribute of each marker.

I've tried

import matplotlib.markers as mmark list_mak = [mmark.MarkerStyle('.'),mmark.MarkerStyle(','),mmark.MarkerStyle('o')] list_lab = ['Marker 1','Marker 2','Marker 3']  plt.legend(list_mak,list_lab) 

But :

1) The MarkerStyle class doesn't support color information

2) I get the warning :

UserWarning: Legend does not support <matplotlib.markers.MarkerStyle object at 0x7fca640c44d0> instances. A proxy artist may be used instead. 

But how can I define a proxy artist based on a marker ?

Thanks for your help !

like image 675
Covich Avatar asked Nov 20 '17 12:11

Covich


2 Answers

Following the example given in the legend guide, you can use a Line2D object instead of a marker object.

The only difference to the example given in the guide is you want to set linestyle='None'

import matplotlib.lines as mlines import matplotlib.pyplot as plt  blue_star = mlines.Line2D([], [], color='blue', marker='*', linestyle='None',                           markersize=10, label='Blue stars') red_square = mlines.Line2D([], [], color='red', marker='s', linestyle='None',                           markersize=10, label='Red squares') purple_triangle = mlines.Line2D([], [], color='purple', marker='^', linestyle='None',                           markersize=10, label='Purple triangles')  plt.legend(handles=[blue_star, red_square, purple_triangle])  plt.show() 

enter image description here

like image 130
tmdavison Avatar answered Oct 04 '22 13:10

tmdavison


You could subclass the HandlerBase to create a handler from a tuple of (color, marker).

import matplotlib.pyplot as plt from matplotlib.legend_handler import HandlerBase  list_color  = ["c", "gold", "crimson"] list_mak    = ["d","s","o"] list_lab    = ['Marker 1','Marker 2','Marker 3']  ax = plt.gca()  class MarkerHandler(HandlerBase):     def create_artists(self, legend, tup,xdescent, ydescent,                         width, height, fontsize,trans):         return [plt.Line2D([width/2], [height/2.],ls="",                        marker=tup[1],color=tup[0], transform=trans)]   ax.legend(list(zip(list_color,list_mak)), list_lab,            handler_map={tuple:MarkerHandler()})   plt.show() 

enter image description here

like image 27
ImportanceOfBeingErnest Avatar answered Oct 04 '22 15:10

ImportanceOfBeingErnest