python - 如何在图例中有多边形

Question

我有一个包含 4 个元素的图表:蓝色方块、蓝色三角形、红色方块和红色三角形，
我正在尝试将这 4 个项目添加到图例中。
当我向图例添加多边形补丁时，它显示为矩形，即使补丁是三角形的
我怎样才能让它工作？

legend_elements = [patches.RegularPolygon((4, 4), 3, 0.5, np.pi / 4,label="Triangle")]
plt.legend(handles=legend_elements)

Answer 1

您可以使用线路 markers指示多边形:

from matplotlib import pyplot as plt
from matplotlib.lines import Line2D

handles = [Line2D([0], [0], linestyle='none', mfc='blue', mec='blue', marker='s', label='blue square'),
           Line2D([0], [0], linestyle='none', mfc='blue', mec='blue', marker='^', label='blue triangle'),
           Line2D([0], [0], linestyle='none', mfc='red', mec='red', marker='s', label='red square'),
           Line2D([0], [0], linestyle='none', mfc='red', mec='red', marker='^', label='red triangle')]
plt.legend(handles=handles)
plt.show()