创建一个矩形双色矩形的matplotlib mpatches图形图例

Question

我需要为matplotlib图形中的图例创建一个补丁，该图形的手柄盒是一个带两种颜色的矩形。像这样的事情，我用油漆应付，切割和着色：

你能告诉我如何以及从哪里开始？ 谢谢。

Answer 1

你当然会从matplotlib legend guide开始;更具体地在关于implementing a custom handler的部分。

阅读定制的传说其他一些问题，如

• Completely custom legend in Matplotlib, Python
• using mpatches.Patch for a custom legend
• Insert image in matplotlib legend
• Replace Matplotlib legend's labels with image
• How to show shapefile label in python basemap legend?

也可能有所帮助。

这里你想在图例中放置两个矩形。因此，在自定义Handler类中，您可以创建这些类并将它们添加到handlebox。

import matplotlib.pyplot as plt 

class Handler(object): 
    def __init__(self, color): 
     self.color=color 
    def legend_artist(self, legend, orig_handle, fontsize, handlebox): 
     x0, y0 = handlebox.xdescent, handlebox.ydescent 
     width, height = handlebox.width, handlebox.height 
     patch = plt.Rectangle([x0, y0], width, height, facecolor='blue', 
            edgecolor='k', transform=handlebox.get_transform()) 
     patch2 = plt.Rectangle([x0+width/2., y0], width/2., height, facecolor=self.color, 
            edgecolor='k', transform=handlebox.get_transform()) 
     handlebox.add_artist(patch) 
     handlebox.add_artist(patch2) 
     return patch 


plt.gca() 

handles = [plt.Rectangle((0,0),1,1) for i in range(4)] 
colors = ["limegreen", "red", "gold", "blue"] 
hmap = dict(zip(handles, [Handler(color) for color in colors])) 

plt.legend(handles=handles, labels=colors, handler_map=hmap) 

plt.show()