added radio buttons

jdh2358 · jdh2358 · commit ff2cdf570edc · 2005-06-14T20:33:58.000Z
svn path=/trunk/matplotlib/; revision=1452
diff --git a/examples/widgets/radio_buttons.py b/examples/widgets/radio_buttons.py
@@ -9,24 +9,27 @@
 l, = ax.plot(t, s0, lw=2, color='red')
 subplots_adjust(left=0.3)
 
-rax = axes([0.05, 0.4, 0.175, 0.175])
-radio = RadioButtons(rax, ('button 1', 'button 2', 'button 3'))
-
+rax = axes([0.05, 0.7, 0.175, 0.175])
+radio = RadioButtons(rax, ('2 Hz', '4 Hz', '8 Hz'))
+def hzfunc(label):
+    hzdict = {'2 Hz':s0, '4 Hz':s1, '8 Hz':s2}
+    ydata = hzdict[label]
+    l.set_ydata(ydata)
+    draw()
+radio.on_clicked(hzfunc)
 
-def func(label):
-    if label=='button 1':
-        ydata = s0
-        color = 'red'
-    elif label=='button 2':
-        ydata = s1
-        color='blue'        
-    elif label=='button 3':
-        ydata = s2
-        color='green'
+rax = axes([0.05, 0.4, 0.175, 0.175])
+radio = RadioButtons(rax, ('red', 'blue', 'green'))
+def colorfunc(label):
+    l.set_color(label)
+    draw()
+radio.on_clicked(colorfunc)
 
-    l.set_ydata(ydata)
-    l.set_color(color)
+rax = axes([0.05, 0.1, 0.175, 0.175])
+radio = RadioButtons(rax, ('-', '--', '-.', 'steps', ':'))
+def stylefunc(label):
+    l.set_linestyle(label)
     draw()
-radio.on_clicked(func)
+radio.on_clicked(stylefunc)
     
 show()
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
@@ -1,5 +1,10 @@
 """
 GUI Neutral widgets
+
+All of these widgets require you to predefine an Axes instance and
+pass that as the first arg.  matplotlib doesn't try to be too smart in
+layout -- you have to figure out how wide and tall you want your Axes
+to be to accommodate your widget.
 """
 
 from matplotlib.mlab import linspace, dist
