matplotlib: faire correspondre la couleur du texte de la légende avec le symbole dans le nuage de points

Question

J'ai créé un nuage de points avec 3 couleurs différentes et je veux faire correspondre la couleur du symbole et du texte dans la légende.

Une belle solution existe pour le cas des parcelles de ligne:

leg = ax.legend() 

# change the font colors to match the line colors: 
for line,text in zip(leg.get_lines(), leg.get_texts()): 
    text.set_color(line.get_color()) 

Cependant, les couleurs de courbe de dispersion ne sont pas accessibles par get_lines() .Pour le cas de 3 couleurs, je pense que je peux définir manuellement les couleurs de texte de un par-un en utilisant par exemple. text.set_color('r'). Mais j'étais curieux de savoir si cela peut être fait automatiquement en lignes. Merci!

Answer 1

Cela semble compliqué mais ne vous donne ce que vous voulez. Les suggestions sont les bienvenues. J'utilise ax.get_legend_handles_labels() pour obtenir les marqueurs et utiliser tuple(handle.get_facecolor()[0]) pour obtenir le tuple couleur matplotlib. Fait un exemple avec un diagramme de dispersion très simple comme ceci:

Edit:

Comme ImportanceOfBeingErnest a souligné dans son answer:

1. leg.legendHandles retournera la légende poignées;
2. La liste, au lieu de tuple, peut être utilisée pour affecter la couleur matplotlib.

Les codes sont simplifiées:

import matplotlib.pyplot as plt 
from numpy.random import rand 


fig, ax = plt.subplots() 
for color in ['red', 'green', 'blue']: 
    x, y = rand(2, 10) 
    ax.scatter(x, y, c=color, label=color) 

leg = ax.legend() 
for handle, text in zip(leg.legendHandles, leg.get_texts()): 
    text.set_color(handle.get_facecolor()[0]) 

plt.show() 

Ce que je suis est:

Answer 2

Les diagrammes de dispersion ont une facecolor et un edgecolor. Le gestionnaire de légende pour un nuage de points est un PathCollection.

vous pouvez donc en boucle sur les poignées de légende et définir la couleur du texte à l'facecolor de la légende gérer

for h, t in zip(leg.legendHandles, leg.get_texts()): 
    t.set_color(h.get_facecolor()[0]) 

Code complet:

import matplotlib.pyplot as plt 
import numpy as np 

fig, ax = plt.subplots() 
for i in range(3): 
    x,y = np.random.rand(2, 20) 
    ax.scatter(x, y, label="Label {}".format(i)) 

leg = ax.legend() 

for h, t in zip(leg.legendHandles, leg.get_texts()): 
    t.set_color(h.get_facecolor()[0]) 

plt.show()