如何为Matplotlib图例添加颜色代码？极坐标雷达图图例问题求助

刚接触Matplotlib时，极坐标图的图例确实容易踩坑——别担心，你遇到的问题很常见，咱们一步步来解决。

你提到“添加对应颜色的句柄时图例就消失”，大概率是因为句柄类型不匹配：极坐标图里绘制的是线条（Line2D对象），如果手动创建补丁（Patch）作为句柄，Matplotlib无法正确关联，就会导致图例显示异常甚至消失。

正确的做法：关联绘图对象与图例 #

核心思路是：保存每次绘图（plot/fill）返回的对象，用这些对象作为图例的句柄。下面结合你的代码片段，给出完整的可运行示例：

import matplotlib.pyplot as plt
import numpy as np

# 1. 设置全局轴样式（你的原有代码）
plt.rc('axes', linewidth=0.5, edgecolor="#888888")

# 2. 准备示例数据
categories = ['速度', '力量', '敏捷', '耐力', '防御']
n_categories = len(categories)
# 生成极坐标角度（注意要闭合图形的话，最后要重复第一个角度）
theta = np.linspace(0, 2*np.pi, n_categories, endpoint=False)
# 两组示例数据
data1 = [8, 6, 9, 7, 5]
data2 = [5, 9, 6, 8, 7]
# 对应颜色
colors = ['#ff7f0e', '#1f77b4']

# 3. 创建极坐标图（你的原有代码）
ax = plt.subplot(111, polar=True)

# 4. 绘制雷达线并保存绘图对象
# 绘制第一组数据，返回Line2D对象
line1, = ax.plot(theta, data1, color=colors[0], linewidth=2)
# 绘制第二组数据
line2, = ax.plot(theta, data2, color=colors[1], linewidth=2)

# 5. 设置轴标签（可选，优化雷达图显示）
ax.set_xticks(theta)
ax.set_xticklabels(categories)

# 6. 添加图例——用保存的Line2D对象作为句柄
# 方式1：直接用plot时指定的label参数
# ax.legend(loc='lower left')

# 方式2：手动指定句柄和标签（适合需要自定义标签的场景）
ax.legend(
    handles=[line1, line2],
    labels=['玩家A', '玩家B'],
    loc='lower left',
    bbox_to_anchor=(-0.1, 0)  # 可选，微调图例位置
)

plt.show()

关键细节解释 #

• 为什么要加逗号？ax.plot()返回的是一个包含Line2D对象的列表，line1, = ax.plot(...)是解包取出单个对象，确保图例能正确识别。
• 如果是填充雷达图（ax.fill()），同样要保存返回的PolyCollection对象，比如：fill1 = ax.fill(theta, data1, color=colors[0], alpha=0.3)，然后把fill1加入handles列表。
• 极坐标的图例位置loc参数和直角坐标略有不同，lower left对应极坐标的左下角区域，用bbox_to_anchor可以更灵活地调整位置。

排查你原有代码的问题 #

如果你之前是手动创建句柄（比如plt.Rectangle），那替换成绘图时返回的Line2D/PolyCollection对象就能解决图例消失的问题。

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