matplotlib check_buttons 框颜色

Question

我正在使用 check_buttons 小部件，如示例 ( http://matplotlib.org/examples/widgets/check_buttons.html )

在具有多条线的绘图上，很难从文本中知道哪个复选框与哪条绘制的线对应，至少在单击该框并且用户查看是否可以判断哪条线消失之前是这样。

有谁知道复选框的背景颜色是否可以与它影响的线相同......类似于图例？

Answer 1

您可以按如下方式设置小部件框颜色，

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

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, visible=False, lw=2)
l1, = ax.plot(t, s1, lw=2)
l2, = ax.plot(t, s2, lw=2)
plt.subplots_adjust(left=0.2)

rax = plt.axes([0.05, 0.4, 0.1, 0.15])
check = CheckButtons(rax, ('2 Hz', '4 Hz', '6 Hz'), (False, True, True))

#Define colours for rectangles and set them
c = ['b', 'g', 'r']    
[rec.set_facecolor(c[i]) for i, rec in enumerate(check.rectangles)]


def func(label):
    if label == '2 Hz': l0.set_visible(not l0.get_visible())
    elif label == '4 Hz': l1.set_visible(not l1.get_visible())
    elif label == '6 Hz': l2.set_visible(not l2.get_visible())
    plt.draw()
check.on_clicked(func)

plt.show()

检查按钮面板将每个复选框作为一个 matplotlib.patches.Rectangle 对象，可以是 customised根据需要。