"""

=============

Check Buttons

=============

Turning visual elements on and off with check buttons.

This program shows the use of 'Check Buttons' which is similar to

check boxes. There are 3 different sine waves shown and we can choose which

waves are displayed with the check buttons.

"""

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, color='k', label='2 Hz')

l1, = ax.plot(t, s1, lw=2, color='r', label='4 Hz')

l2, = ax.plot(t, s2, lw=2, color='g', label='6 Hz')

plt.subplots_adjust(left=0.2)

lines = [l0, l1, l2]

# Make checkbuttons with all plotted lines with correct visibility

rax = plt.axes([0.05, 0.4, 0.1, 0.15])

labels = [str(line.get_label()) for line in lines]

visibility = [line.get_visible() for line in lines]

check = CheckButtons(rax, labels, visibility)

def func(label):

index = labels.index(label)

lines[index].set_visible(not lines[index].get_visible())

plt.draw()

check.on_clicked(func)

plt.show()

#############################################################################

#

# .. admonition:: References

#

# The use of the following functions, methods, classes and modules is shown

# in this example:

#

# - `matplotlib.widgets.CheckButtons`