如何修改第三方包返回的matplotlib.figure对象的图例位置？

核心问题分析 #

第三方程序创建的图例可能未与坐标轴的handles/labels绑定，导致ax.get_legend_handles_labels()返回空列表。无需重新创建图例，直接定位已有图例对象修改即可。

步骤1：定位已存在的图例对象 #

根据第三方程序的实现方式，从以下途径获取图例：

• 从figure对象的图例集合获取（适用于直接在figure上创建的图例）：

  # 获取figure上的第一个图例（若仅存在一个）
  lgd = f.legends[0]
  

• 从坐标轴对象获取（适用于绑定到轴的图例）：

  ax = f.axes[0]
  lgd = ax.get_legend()
  

• 极端情况：若上述方法都找不到，遍历figure所有子对象筛选图例：

  from matplotlib.legend import Legend
  
  lgd = None
  for child in f.get_children():
      if isinstance(child, Legend):
          lgd = child
          break
  

步骤2：修改图例位置与边界框 #

拿到图例对象后，直接调用方法调整参数，不要重新调用ax.legend()（这会生成新图例，覆盖原有或导致空图例）：

# 设置图例锚点到坐标轴右上角外侧，位置设为左上对齐
lgd.set_bbox_to_anchor((1.02, 1))
lgd.set_loc('upper left')

# 可选：调整边框样式
lgd.set_frame_on(True)  # 显示边框
lgd.set_edgecolor('gray')  # 设置边框颜色

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