我正在尝试创建具有三个参数变化的多维函数的交互式matplotlib图.问题在于参数可以在很大的范围内变化,因此我宁愿不使用滑块,而直接键入想要的值.基本上,我想重新创建下面的规范示例,在该示例中,我希望使用文本框(而不是滑块)在其中输入参数

import numpy as np

import matplotlib.pyplot as plt

from matplotlib.widgets import Slider, Button, RadioButtons

fig, ax = plt.subplots()

plt.subplots_adjust(bottom=0.35)

t = np.arange(0.0, 1.0, 0.001)

a0 = 5

f0 = 3

s = a0*np.sin(2*np.pi*f0*t)

l, = plt.plot(t,s, lw=2, color='red')

plt.axis([0, 1, -10, 10])

ax.plot(t,t)

axcolor = 'lightgoldenrodyellow'

axamp = plt.axes([0.25, 0.25, 0.65, 0.03], axisbg=axcolor)

axfreq = plt.axes([0.25, 0.2, 0.65, 0.03], axisbg=axcolor)

sfreq = Slider(axfreq, 'Freq', 0.1, 30.0, valinit=f0)

samp = Slider(axamp, 'Amp', 0.1, 10.0, valinit=a0)

def update(val):

amp = samp.val

freq = sfreq.val

l.set_ydata(amp*np.sin(2*np.pi*freq*t))

fig.canvas.draw_idle()

sfreq.on_changed(update)

samp.on_changed(update)

resetax = plt.axes([0.8, 0.025, 0.1, 0.04])

button = Button(resetax, 'Reset', color=axcolor, hovercolor='0.975')

def reset(event):

sfreq.reset()

samp.reset()

button.on_clicked(reset)

plt.show()

最佳答案

您可以将GUI面板添加到图形窗口.这是一个使用Qt4Agg后端,并将QDockWidget添加到主图形窗口,然后可以将QWidgets添加到停靠窗口的示例.

import numpy as np

import matplotlib

matplotlib.use("Qt4Agg") # This program works with Qt only

import pylab as pl

fig, ax1 = pl.subplots()

t = np.linspace(0, 10, 200)

line, = ax1.plot(t, np.sin(t))

### control panel ###

from PyQt4 import QtGui

from PyQt4 import QtCore

from PyQt4.QtCore import Qt

def update():

freq = float(textbox.text())

y = np.sin(2*np.pi*freq*t)

line.set_data(t, y)

fig.canvas.draw_idle()

root = fig.canvas.manager.window

panel = QtGui.QWidget()

hbox = QtGui.QHBoxLayout(panel)

textbox = QtGui.QLineEdit(parent = panel)

textbox.textChanged.connect(update)

hbox.addWidget(textbox)

panel.setLayout(hbox)

dock = QtGui.QDockWidget("control", root)

root.addDockWidget(Qt.BottomDockWidgetArea, dock)

dock.setWidget(panel)

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

pl.show()

这是屏幕：