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)
s3 = np.sin(8*np.pi*t)

fig, axes = plt.subplot_mosaic(
    [['main'], ['buttons']],
    height_ratios=[8, 1],
    layout="constrained",
)

l0, = axes['main'].plot(t, s0, lw=2, label='2 Hz')
l1, = axes['main'].plot(t, s1, lw=2, label='4 Hz')
l2, = axes['main'].plot(t, s2, lw=2, label='6 Hz')
l3, = axes['main'].plot(t, s3, lw=2, label='8 Hz')
axes['main'].set_xlabel('Time (s)')
axes['main'].set_ylabel('Amplitude')

lines_by_label = {l.get_label(): l for l in [l0, l1, l2, l3]}

axes['buttons'].set_facecolor('0.95')
check = CheckButtons(
    axes['buttons'],
    labels=lines_by_label.keys(),
    actives=[l.get_visible() for l in lines_by_label.values()],
    layout='horizontal'
)

def callback(label):
    ln = lines_by_label[label]
    ln.set_visible(not ln.get_visible())
    fig.canvas.draw_idle()

check.on_clicked(callback)