我想创建一个集成ode模型的脚本,这样我就可以更改参数之一,并查看系统对此更改的响应。例如,如果有一个Lotka-Volterra模型(如本例所示):
import numpy as np
from scipy import integrate
a = 1.
b = 0.1
c = 1.5
d = 0.75
def dX_dt(X, t=0):
""" Return the growth rate of fox and rabbit populations. """
return array([ a*X[0] - b*X[0]*X[1] ,
-c*X[1] + d*b*X[0]*X[1] ])
t = np.linspace(0, 15, 1000) # time
X0 = np.array([10, 5]) # initials conditions: 10 rabbits and 5 foxes
X, infodict = integrate.odeint(dX_dt, X0, t, full_output=True)
我想为参数a
和创建一个滑块c
,如matplotlib的slider_demo或任何其他工具一样。该图应显示一个始终跨度的特定时间窗口[t_current - delta_t ,t_current]
。因此,通过更改参数的滑块,我将能够连续探索参数空间。
怎么做?
您已经掌握了所有内容,基本上只是更改了小部件示例中的update方法,以使用基于滑块的新值重新计算dX_dt的积分,然后使用它来设置y行值。代码如下:
import numpy as np
from scipy import integrate
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button, RadioButtons
b = 0.1
d = 0.75
a=1
c=1.5
def dX_dt(X, t=0, a=1, c=1.5):
""" Return the growth rate of fox and rabbit populations. """
return np.array([ a*X[0] - b*X[0]*X[1] ,
-c*X[1] + d*b*X[0]*X[1] ])
t = np.linspace(0, 15, 1000) # time
X0 = np.array([10, 5]) # initials conditions: 10 rabbits and 5 foxes
fig, ax = plt.subplots()
plt.subplots_adjust(left=0.25, bottom=0.25)
l1, l2 = plt.plot(t, integrate.odeint(dX_dt, X0, t, (a, c)))
axcolor = 'black'
ax_a = plt.axes([0.25, 0.1, 0.65, 0.03], axisbg=axcolor)
ax_c = plt.axes([0.25, 0.15, 0.65, 0.03], axisbg=axcolor)
sa = Slider(ax_a, 'a', 0.1, 10.0, valinit=1)
sc = Slider(ax_c, 'c', 0.1, 10.0, valinit=1.5)
def update(val):
a = sa.val
c = sc.val
x = integrate.odeint(dX_dt, X0, t, (a, c))
l1.set_ydata(x[:,0])
l2.set_ydata(x[:,1])
fig.canvas.draw_idle()
sa.on_changed(update)
sc.on_changed(update)
plt.show()
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句