用两个滑块交互式绘制matplotlib图
我用matplotlib创建了一些图,这取决于8个变量。 我想研究当我改变其中一些情节时情节如何改变。 我创建了一些调用matplotlib的脚本,并生成不同的快照,稍后我将其转换为电影,但这并不坏,但有点笨拙。
我想知道是否可以通过键盘按键来与绘图再生相互作用,以增加/减少某些变量的值,并立即查看绘图如何变化。
什么是最好的方法呢?
另外,如果您可以用两个滑块将我指向有趣的链接或与一个情节示例的链接?
除了@triplepoint提到的内容之外,还要看看滑块小部件。
matplotlib示例页面上有一个示例。 这是一个图形滑动条而不是键盘绑定,但它对于你想要做的事情非常有效。
还要注意,要保证滑块和按钮保持响应并且不被垃圾收集,对象( amp_slider
, freq_slider
等)的引用应由您自己维护。
(我正在制作这个社区wiki,因为我只是从这个例子中复制粘贴),这个特殊的例子教会了一些不良的习惯(例如from pylab import *
),但是它得到了重点。使用pylab
。)
from numpy import pi, sin
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button, RadioButtons
def signal(amp, freq):
return amp * sin(2 * pi * freq * t)
axis_color = 'lightgoldenrodyellow'
fig = plt.figure()
ax = fig.add_subplot(111)
# Adjust the subplots region to leave some space for the sliders and buttons
fig.subplots_adjust(left=0.25, bottom=0.25)
t = np.arange(0.0, 1.0, 0.001)
amp_0 = 5
freq_0 = 3
# Draw the initial plot
# The 'line' variable is used for modifying the line later
[line] = ax.plot(t, signal(amp_0, freq_0), linewidth=2, color='red')
ax.set_xlim([0, 1])
ax.set_ylim([-10, 10])
# Add two sliders for tweaking the parameters
# Define an axes area and draw a slider in it
amp_slider_ax = fig.add_axes([0.25, 0.15, 0.65, 0.03], axisbg=axis_color)
amp_slider = Slider(amp_slider_ax, 'Amp', 0.1, 10.0, valinit=amp_0)
# Draw another slider
freq_slider_ax = fig.add_axes([0.25, 0.1, 0.65, 0.03], axisbg=axis_color)
freq_slider = Slider(freq_slider_ax, 'Freq', 0.1, 30.0, valinit=freq_0)
# Define an action for modifying the line when any slider's value changes
def sliders_on_changed(val):
line.set_ydata(signal(amp_slider.val, freq_slider.val))
fig.canvas.draw_idle()
amp_slider.on_changed(sliders_on_changed)
freq_slider.on_changed(sliders_on_changed)
# Add a button for resetting the parameters
reset_button_ax = fig.add_axes([0.8, 0.025, 0.1, 0.04])
reset_button = Button(reset_button_ax, 'Reset', color=axis_color, hovercolor='0.975')
def reset_button_on_clicked(mouse_event):
freq_slider.reset()
amp_slider.reset()
reset_button.on_clicked(reset_button_on_clicked)
# Add a set of radio buttons for changing color
color_radios_ax = fig.add_axes([0.025, 0.5, 0.15, 0.15], axisbg=axis_color)
color_radios = RadioButtons(color_radios_ax, ('red', 'blue', 'green'), active=0)
def color_radios_on_clicked(label):
line.set_color(label)
fig.canvas.draw_idle()
color_radios.on_clicked(color_radios_on_clicked)
plt.show()
Matplotlib有一些相当不错的gui功能。 在matplotlib的源码tarball中有一些文档示例,位于/ examples / user_interfaces和matplotlib> / examples / event_handling中。 特别是在keyhandling上是:http://matplotlib.sourceforge.net/examples/event_handling/keypress_demo.html
我做了一些类似于你想要的东西:
import numpy as np
import pylab
class plotter:
def __init__(self, initial_values):
self.values
self.fig = pylab.figure()
pylab.gray()
self.ax = self.fig.add_subplot(111)
self.draw()
self.fig.canvas.mpl_connect('key_press_event',self.key)
def draw(self):
im = your_function(self.values)
pylab.show()
self.ax.imshow(im)
def key(self, event):
if event.key=='right':
self.values = modify()
elif event.key == 'left':
self.values = modify()
self.draw()
self.fig.canvas.draw()
我正在使用这个来移位通过在按键上堆叠显示不同的图像,但是您应该能够在给定键盘输入的情况下通过逻辑来修改您的值。
如果你想做的事情就像拥有用户输入值一样,我认为这些例子有对话框的选项,但是如果你只想增加/减少一些变量,那么只用这种方式为它们定义键盘对可能会工作得很好
使用waitforbuttonpress(timeout=0.001)
然后情节会看到你的鼠标滴答声。