注意
转到末尾以下载完整的示例代码。
复选按钮#
使用复选按钮打开和关闭视觉元素。
此程序显示了 CheckButtons
的用法,它类似于复选框。这里显示了 3 种不同的正弦波,我们可以使用复选按钮选择要显示的波形。
可以使用 *check_props*、*frame_props* 和 *label_props* 参数设置复选按钮的样式。这些参数都接受一个字典,其中键是艺术家属性名称,值是设置列表,其长度与按钮的数量相匹配。
import matplotlib.pyplot as plt
import numpy as np
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='black', label='1 Hz')
l1, = ax.plot(t, s1, lw=2, color='red', label='2 Hz')
l2, = ax.plot(t, s2, lw=2, color='green', label='3 Hz')
lines_by_label = {l.get_label(): l for l in [l0, l1, l2]}
line_colors = [l.get_color() for l in lines_by_label.values()]
# Make checkbuttons with all plotted lines with correct visibility
rax = ax.inset_axes([0.0, 0.0, 0.12, 0.2])
check = CheckButtons(
ax=rax,
labels=lines_by_label.keys(),
actives=[l.get_visible() for l in lines_by_label.values()],
label_props={'color': line_colors},
frame_props={'edgecolor': line_colors},
check_props={'facecolor': line_colors},
)
def callback(label):
ln = lines_by_label[label]
ln.set_visible(not ln.get_visible())
ln.figure.canvas.draw_idle()
check.on_clicked(callback)
plt.show()
脚本的总运行时间:(0 分钟 1.091 秒)