scipy.integrate.ode,函数返回元组问题

时间:2016-11-06 22:09:23

标签: numpy scipy ode integrator

所以我编写了一个函数来使用特定的求解器类型来集成一对ODE系统。由于某些原因,我无法理解,我编写的用于定义ODE的RHS的函数在积分函数内运行时返回一个元组(或者错误指示),但每当我测试时它单独返回的数组,数组似乎......好......和数组,应该可以正常工作。

def solve_decay_system(t, u_0, solver_type="dopri5", K_1=0.0, K_2=0.0, K_3=0.0):

    def decay(u, K_1 = K_1, K_2=K_2, K_3=K_3):

        dydt = numpy.array([-K_1 * u[0], K_1 * u[0] - K_2 * u[1], K_2 * u[1] - K_3*u[2], K_3*u[2]])
        dydt = numpy.array(dydt)
        print dydt, type(dydt)
        return dydt
        #return [1, 2, 3, 4]

    u = numpy.empty((4, t.size))
    u_0 = numpy.array([1.0, 0.0, 0.0, 0.0])
    u[0, 0] = u_0[0]
    u[1, 0] = u_0[1]
    u[2, 0] = u_0[2]
    u[3, 0] = u_0[3]

    integrator = integrate.ode(decay)
    integrator.set_integrator(solver_type)
    integrator.set_initial_value(u[:, 0])
    integrator.set_f_params(u, K_1, K_2, K_3)

    for (n, t_n) in enumerate(t[1:]):
        print n
        integrator.integrate(t_n)
        if not integrator.successful():
            break

        u[:, n + 1] = integrator.y


    return u

有关如何解决此问题的任何建议都非常有用。以下是结果错误,供参考:

ValueError                                Traceback (most recent call last)
<ipython-input-230-8f9be46c3d95> in <module>()
----> 1 solve_decay_system(t, u_0, solver_type="dopri5", K_1=0.0, K_2=0.0,     K_3=0.0)

<ipython-input-229-a05ddca0f334> in solve_decay_system(t, u_0, solver_type,     K_1, K_2, K_3)
     24     for (n, t_n) in enumerate(t[1:]):
     25         print n
---> 26         integrator.integrate(t_n)
     27         if not integrator.successful():
     28             break

C:\Users\Ben\Anaconda2\lib\site-packages\scipy\integrate\_ode.pyc in     integrate(self, t, step, relax)
    409         except SystemError:
    410             # f2py issue with tuple returns, see ticket 1187.
--> 411             raise ValueError('Function to integrate must not return     a tuple.')
    412 
    413         return self._y

ValueError: Function to integrate must not return a tuple.

1 个答案:

答案 0 :(得分:1)

派生函数期望(time, state,...)作为第一个参数,

def decay(t, u, K_1 = K_1, K_2=K_2, K_3=K_3):

即使ODE是自治的。

状态不是参数,尤其是如果你将状态列表作为参数。

  integrator.set_f_params(K_1, K_2, K_3)

通过这些更改,您的代码应该运行。 (最多n=8。)

即使您只是返回一个列表

def decay(t, u, K_1 = K_1, K_2=K_2, K_3=K_3):
    return  [-K_1 * u[0], K_1 * u[0] - K_2 * u[1], K_2 * u[1] - K_3*u[2], K_3*u[2]]