Dans Matplotlib, j'essaie de créer une légende avec des "marqueurs" colorés comme celui-ci:
celui-ci a été créé avec la fonction scatter
, mais n’est pas adapté à mon graphique. Je voudrais produire une légende "à partir de zéro", sans données associées . La couleur est importante, et devrait donc être un attribut de chaque marqueur.
J'ai essayé
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)
Mais :
1) La classe MarkerStyle
ne prend pas en charge les informations de couleur.
2) je reçois l'avertissement:
UserWarning: Legend does not support <matplotlib.markers.MarkerStyle object at 0x7fca640c44d0> instances.
A proxy artist may be used instead.
Mais comment puis-je définir un artiste proxy basé sur un marqueur?
Merci de votre aide !
Suivant l'exemple donné dans le guide legend , vous pouvez utiliser un objet Line2D
au lieu d'un objet marker
.
La seule différence par rapport à l'exemple donné dans le guide est que vous souhaitez définir 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()
Vous pouvez sous-classer la HandlerBase
pour créer un gestionnaire à partir d'un tuple de (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()