如何在Matplotlib复选按钮文本旁添加对应线条颜色标识？

这问题我做可视化时也碰到过，用图例跟着复选按钮确实容易出现窗口缩放移位的问题，其实咱们可以直接把线条颜色整合到复选按钮组件里，下面分享两种亲测有效的方法：

方法一：给复选按钮标签添加颜色方块（直观视觉对应） #

Matplotlib的文本支持HTML格式，咱们可以给每个复选按钮的标签前加上和对应线条同色的方块，这样标签本身就带颜色标识，完全不会随窗口移位。

示例代码：

import matplotlib.pyplot as plt
from matplotlib.widgets import CheckButtons
import numpy as np

# 生成测试数据
t = np.arange(0.0, 2.0, 0.01)
s0 = np.sin(2*np.pi*t)
s1 = np.sin(4*np.pi*t)
s2 = np.sin(6*np.pi*t)

# 创建画布和子图
fig, ax = plt.subplots()
l0, = ax.plot(t, s0, label='sin(2πt)', color='#1f77b4')
l1, = ax.plot(t, s1, label='sin(4πt)', color='#ff7f0e')
l2, = ax.plot(t, s2, label='sin(6πt)', color='#2ca02c')
lines = [l0, l1, l2]

# 准备带颜色方块的标签：用HTML span设置颜色，加上方块符号
labels = []
for line in lines:
    color = line.get_color()
    # ■ 是方块符号，也可以用●等其他符号
    colored_label = f"<span style='color:{color}'>■</span> {line.get_label()}"
    labels.append(colored_label)

# 创建复选按钮区域
rax = plt.axes([0.05, 0.4, 0.15, 0.3])
check = CheckButtons(rax, labels, [True, True, True])

# 控制线条显示/隐藏的函数
def func(label):
    # 提取原始标签（去掉颜色方块部分）
    original_label = label.split(' ')[-1]
    # 找到对应线条的索引
    index = [line.get_label() for line in lines].index(original_label)
    lines[index].set_visible(not lines[index].get_visible())
    plt.draw()

check.on_clicked(func)

plt.show()

注意：func函数里需要把标签里的颜色方块部分去掉，提取原始线条标签来匹配索引，不然会找不到对应线条。

方法二：直接修改复选按钮的背景颜色 #

如果觉得标签加方块不够简洁，咱们可以把每个复选按钮的背景色改成对应线条的颜色，让按钮本身就代表线条的颜色，选中状态可以调整颜色深浅来区分。

示例代码：

import matplotlib.pyplot as plt
from matplotlib.widgets import CheckButtons
import numpy as np

# 生成测试数据和创建线条部分和方法一一致
t = np.arange(0.0, 2.0, 0.01)
s0 = np.sin(2*np.pi*t)
s1 = np.sin(4*np.pi*t)
s2 = np.sin(6*np.pi*t)

fig, ax = plt.subplots()
l0, = ax.plot(t, s0, label='sin(2πt)', color='#1f77b4')
l1, = ax.plot(t, s1, label='sin(4πt)', color='#ff7f0e')
l2, = ax.plot(t, s2, label='sin(6πt)', color='#2ca02c')
lines = [l0, l1, l2]
labels = [line.get_label() for line in lines]

# 创建复选按钮区域
rax = plt.axes([0.05, 0.4, 0.15, 0.3])
check = CheckButtons(rax, labels, [True, True, True])

# 给每个复选按钮设置对应线条的背景色
for line, button in zip(lines, check.buttons):
    # 设置按钮背景色为线条颜色
    button.rect.set_facecolor(line.get_color())
    # 可选：设置选中时的颜色（浅一点的同色系，加透明度）
    button.active_rect.set_facecolor(line.get_color() + '80')

# 控制函数：直接用原始标签匹配
def func(label):
    index = labels.index(label)
    lines[index].set_visible(not lines[index].get_visible())
    plt.draw()

check.on_clicked(func)

plt.show()

这两种方法都完美解决了图例移位的问题，因为颜色标识是复选按钮组件的一部分，会随窗口缩放自动适配位置，你可以根据自己的可视化风格选其中一种~

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