matplotlib: màu văn bản chú thích phù hợp với biểu tượng trong ô phân tán

Question

Tôi đã vẽ một ô phân tán với 3 màu khác nhau và tôi muốn khớp màu của biểu tượng và văn bản trong chú giải.

A nice solution tồn tại đối với trường hợp thửa dòng:

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

Tuy nhiên, màu sắc biểu đồ phân tán không thể được truy cập bởi get_lines() .Đối với trường hợp của 3 màu Tôi nghĩ rằng tôi có thể tự thiết lập các màu sắc văn bản one- bằng cách sử dụng ví dụ. text.set_color('r'). Nhưng tôi đã tò mò nếu nó có thể được thực hiện tự động như là dòng. Cảm ơn!

Answer 1

Điều này có vẻ phức tạp nhưng không cung cấp cho bạn những gì bạn muốn. Gợi ý được hoan nghênh. Tôi sử dụng ax.get_legend_handles_labels() để lấy các điểm đánh dấu và sử dụng tuple(handle.get_facecolor()[0]) để lấy bộ màu matplotlib. Đưa ra một ví dụ với một âm mưu thực sự đơn giản phân tán như thế này:

Edit:

Như ImportanceOfBeingErnest chỉ trong answer mình:

1. leg.legendHandles sẽ trở lại với huyền thoại xử lý;
2. Danh sách, thay vì tuple, có thể được sử dụng để gán màu matplotlib.

Codes được đơn giản hóa như:

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

Những gì tôi nhận được là:

Answer 2

lô Scatter có một facecolor và một edgecolor. Trình xử lý chú giải cho một phân tán là PathCollection.

Vì vậy, bạn vòng qua tay cầm huyền thoại và thiết lập màu chữ cho facecolor của huyền thoại có thể xử lý

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

mã hoàn chỉnh:

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