matplotlib: Übereinstimmung Legend Textfarbe mit Symbol in Streudiagramm

Question

Ich habe ein Streudiagramm mit 3 verschiedenen Farben erstellt und möchte die Farbe des Symbols und den Text in der Legende übereinstimmen.

Ein schöner solution gibt es für den Fall der Liniendiagramme:

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()) 

jedoch Streudiagramm Farben nicht durch get_lines() .Für den Fall von 3 Farben ich glaube, ich manuell den Text zugegriffen werden kann Farben festlegen können ein- by-one mit z. text.set_color('r'). Aber ich war neugierig, ob das automatisch als Zeilen gemacht werden kann. Vielen Dank!

Answer 1

Dies scheint kompliziert, aber gibt Ihnen, was Sie wollen. Vorschläge sind willkommen. Ich verwende ax.get_legend_handles_labels(), um die Marker zu erhalten und tuple(handle.get_facecolor()[0]) zu verwenden, um das Matplotlib-Farbtupel zu erhalten. Hergestellt ein Beispiel mit einem wirklich einfachen Streudiagramm wie folgt aus:

Edit:

Als ImportanceOfBeingErnest in seiner answer zeigte:

1. leg.legendHandles die Legende zurück behandelt;
2. Liste anstelle von Tupel, kann verwendet werden, um Matplotlib-Farbe zuzuweisen.

Codes werden als vereinfacht:

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() 

Was ich ist bekam:

Answer 2

Streudiagramme eine facecolor und eine edgecolor haben. Der Legendenhandler für eine Streuung ist ein PathCollection.

So können Sie Schleife über die Legende Griffe und stellen Sie die Textfarbe auf die facecolor der Legende Griff

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

komplette Code:

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()