如何在图例中有一个多边形

How to have a poligon in the legend

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

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

您可以使用线 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()