matplotlib: mecz kolor tekstu legenda o symbolu w wykres punktowy

Question

Zrobiłem wykres punktowy z 3 różnych kolorach i chcę dopasować kolor symbol i tekst w legendzie.

Miła solution istnieją w przypadku działek linia:

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

Jednak kolory wykres punktowy nie mogą być dostępne przez get_lines() .W przypadku 3 kolorach myślę, że mogę ręcznie ustawić kolory tekstu jedno- osobno przy użyciu np. text.set_color('r'). Ale byłem ciekawy, czy można to zrobić automatycznie jako linie. Dzięki!

Answer 1

to wydaje się skomplikowane, ale daje Ci to, co chcesz. Sugestie są mile widziane. Używam ax.get_legend_handles_labels(), aby uzyskać znaczniki i użyć tuple(handle.get_facecolor()[0]), aby uzyskać krotkę koloru matplotlib. Wykonane przykład z naprawdę prosty wykres punktowy jak ten:

Edit:

Jak ImportanceOfBeingErnest wskazał w Jego answer:

1. leg.legendHandles zwróci obsługuje legendę;
2. Lista, zamiast krotki, może być użyta do przypisania koloru matplotlib.

Kody są uproszczone jak:

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

Co mam to:

Answer 2

Wykresy rozrzutu mają facecolor oraz edgecolor. Obsługa legendy dla rozproszenia to PathCollection.

Więc można pętli nad uchwytami legendy i ustawić kolor tekstu do facecolor legendy obsłużyć

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

Kompletna kod:

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