如何在Matplotlib中创建自定义箭头形状的图例标识

方法一：改进自定义Handler，适配图例容器尺寸 #

你之前的自定义ArrowHandler硬编码了箭头的位置和尺寸，导致和图例的容器尺寸不匹配。应该利用create_artists方法提供的width和height参数动态计算箭头位置，同时匹配图中箭头的垂直向下方向：

from matplotlib.legend_handler import HandlerBase
from matplotlib.patches import FancyArrowPatch

class ArrowHandler(HandlerBase):
    def create_artists(self, legend, orig_handle, xdescent, ydescent, width, height, fontsize, trans):
        # 基于图例项的宽高计算箭头起点和终点，保持垂直向下
        start_x = width / 2
        start_y = height - 2
        end_x = width / 2
        end_y = 2
        # 对齐图中箭头的样式：线条宽度、箭头头部大小
        arrow = FancyArrowPatch((start_x, start_y), (end_x, end_y),
                               arrowstyle='->', color=orig_handle.get_edgecolor(),
                               linewidth=2,  # 和图中线条宽度匹配
                               mutation_scale=10)  # 调整箭头头部大小，按需微调
        # 应用变换，确保箭头在图例中正确定位
        arrow.set_transform(trans)
        return [arrow]

然后定义图例元素时，只需指定颜色即可：

legend_elements = [
    FancyArrowPatch((0,0), (0,0), color='#1b9e77', label='泥石流'),
    FancyArrowPatch((0,0), (0,0), color='#7570b3', label='碎屑崩落')
]

plt.legend(handles=legend_elements, handler_map={FancyArrowPatch: ArrowHandler()})

方法二：更简便的模拟方式——用Line2D实现 #

如果不想编写自定义Handler，可以用带向下三角形标记的线条模拟箭头，视觉效果接近且代码更简洁：

from matplotlib.lines import Line2D

legend_elements = [
    Line2D([0], [0], color='#1b9e77', linewidth=2, marker='v', markersize=8, label='泥石流'),
    Line2D([0], [0], color='#7570b3', linewidth=2, marker='v', markersize=8, label='碎屑崩落')
]

plt.legend(handles=legend_elements)

关键注意事项 #

• 图中使用plt.arrow，图例用FancyArrowPatch，两者样式参数不完全一致，自定义Handler时需对齐线条宽度、箭头头部比例等细节
• mutation_scale参数可调整FancyArrowPatch的箭头头部大小，按需微调即可
• 自定义Handler必须应用trans变换，否则箭头位置会出现偏移

内容的提问来源于stack exchange，提问作者Porter