如何为Matplotlib图形中的三角形标注添加额外图例？

嘿，刚接触Matplotlib的时候，遇到这种自定义元素的图例问题确实容易摸不着头绪，我给你分享两种最实用的方法，直接就能上手：

方法1：用自定义补丁（Patch）创建图例 #

这种方法适合你已经用多边形（比如Polygon）画了三角形标注的场景，核心是把你的三角形转换成图例能识别的补丁对象：

import matplotlib.pyplot as plt
from matplotlib.patches import Polygon

# 先画你的主图（这里用示例数据代替）
x = [1,2,3,4]
y = [2,4,3,5]
plt.plot(x, y, label='折线数据')

# 示例：画一个三角形标注（假设你已经有类似的代码）
triangle_vertices = [(2,4), (2.2,3.8), (1.8,3.8)]  # 三角形的三个顶点
tri_patch = Polygon(triangle_vertices, color='red', alpha=0.5)
plt.gca().add_patch(tri_patch)

# 给三角形创建单独的图例
# 先创建一个和标注样式一致的补丁（不用加到图上，只用来做图例）
legend_tri = Polygon([[0,0],[0,0],[0,0]], color='red', alpha=0.5)  # 顶点随便设，不影响图例显示
plt.legend(handles=[plt.plot([], [], label='折线数据')[0], legend_tri], 
           labels=['折线数据', '峰值标记'])

plt.show()

解释一下：我们创建一个和实际三角形样式完全一致的空补丁（顶点随便设，因为图例里只会显示形状和颜色），然后把它和原有图例元素一起传给plt.legend()的handles参数就行。

方法2：用空绘图命令快速创建图例 #

如果你的三角形是用scatter或者annotate里的标记画的（比如用marker='^'），这种方法更简单：

import matplotlib.pyplot as plt

# 主图示例
x = [1,2,3,4]
y = [2,4,3,5]
plt.plot(x, y, label='折线数据')

# 用scatter画三角形标注（示例）
plt.scatter(2,4, marker='^', color='green', s=100, label='')  # 这里label留空，避免自动进图例

# 给三角形创建图例：画一个空的scatter/plot，只用来定义图例样式
plt.scatter([], [], marker='^', color='green', s=100, label='峰值标记')

# 直接调用legend，它会自动收集所有带label的元素
plt.legend()

plt.show()

这种方法的核心是：画一个没有实际数据（x、y用空列表）的标记图，设置和你实际三角形标注完全一样的样式（标记形状、颜色、大小），然后给它加label，图例就会自动识别这个样式。

新手小提醒 #

• 如果有多种不同的三角形（比如不同颜色、方向），只要重复上面的步骤，给每个样式都创建对应的补丁/空绘图就行
• 搜索相关问题的时候，可以用关键词：matplotlib custom legend for polygon、matplotlib legend for marker annotation，这样更容易找到精准的解决方案

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