如何從頭開始制作彩色標記圖例

Question

在 Matplotlib 中，我試圖用這樣的彩色“標記”制作一個傳奇：

這個是使用scatter函數制作的，但不適合我的情節。 我想“從頭開始”制作一個圖例，沒有相關數據。 顏色很重要，因此應該是每個標記的屬性。

我試過了

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)

但 ：

1. MarkerStyle類不支持顏色信息

2. 我收到警告：

   UserWarning: Legend 不支持 <matplotlib.markers.MarkerStyle object at 0x7fca640c44d0> 實例。 可以使用代理藝術家代替。

但是如何根據標記定義代理藝術家？

Answer 1

按照圖例指南中給出的示例，您可以使用Line2D對象而不是marker對象。

與指南中給出的示例的唯一區別是您要設置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()

Answer 2

您可以HandlerBase以從(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()