matplotlib.github.com/examples/widgets/check_buttons.py at master · coid/matplotlib.github.com · GitHub

26 lines (21 loc) · 715 Bytes

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))

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()

Expand file tree