python - 小方 block 而不是传说中的线条？

Question

要制作自定义图例，我目前使用以下内容:

handles, labels = plt.gca().get_legend_handles_labels()
my_artist = plt.Line2D((0,1),(0,0), color = "blue", linestyle = "-", linewidth = 1)
plt.legend([handle for i,handle in enumerate(handles) if i in display]+[my_artist],
           [label for i,label in enumerate(labels) if i in display]+["My legend"])

它将在图例框中绘制一条蓝线。我想要一个蓝色的小方 block (但比简单的标记大)，而不是一条线。如何做到这一点？

Answer 1

制作一个代理矩形而不是 Line2D，如果您希望它是一个正方形，请对句柄长度大惊小怪(但handlelength和handleheight适用于整个图例):

import matplotlib.pyplot as plt
handles, labels = plt.gca().get_legend_handles_labels()
my_artist = plt.Line2D((0,1),(0,0), color = "blue", linestyle = "-", linewidth = 1)
p = plt.Rectangle((0, 0), 1, 1, fc="b")
plt.legend([handle for i,handle in enumerate(handles) if i in display]+[my_artist, p],
           [label for i,label in enumerate(labels) if i in display]+["Line2D", "Rectangle"], 
           handlelength=0.8, handleheight=0.8)
plt.show()

(几乎直接来自 matplotlib 文档的示例: legend guide 。)