对象太深，无法使用所需的数组-scipy.integrate.odein

Question

我昨天刚开始使用Python，使用scipy.integrate.odeint时出现错误。

我定义了一个函数

def SIR(x, t, beta, gamma, mu, M):

它接受numpy.array对象x、t和M；标量浮动beta、gamma和mu。

M的大小是(60,60)，但我认为这无关紧要。

x和t都是不独立的，其中x.shape是(180,)，t.shape是(5000,)。我试着给它们一个单重维度，使它们分别具有形状(180,1)和(5000,1)，但我仍然得到相同的错误：

In [1]: run measles_age.py
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
/Library/Frameworks/EPD64.framework/Versions/7.3/lib/python2.7/site-packages/IPython/utils/py3compat.py in execfile(fname, *where)
    173             else:
    174                 filename = fname
--> 175             __builtin__.execfile(filename, *where)

/Users/qcaudron/Documents/SIR/measles_age.py in <module>()
    111 
    112 
--> 113         x = integrate.odeint(SIR, x0, t, args=(beta, gamma, mu, M));
    114 
    115 #       plot(t, x);


/Library/Frameworks/EPD64.framework/Versions/7.3/lib/python2.7/site-packages/scipy/integrate/odepack.py in odeint(func, y0, t, args, Dfun, col_deriv, full_output, ml, mu, rtol, atol, tcrit, h0, hmax, hmin, ixpr, mxstep, mxhnil, mxordn, mxords, printmessg)
    141     output = _odepack.odeint(func, y0, t, args, Dfun, col_deriv, ml, mu,
    142                              full_output, rtol, atol, tcrit, h0, hmax, hmin,
--> 143                              ixpr, mxstep, mxhnil, mxordn, mxords)
    144     if output[-1] < 0:
    145         print _msgs[output[-1]]

即使当SIR只返回x时，我也会得到这个错误，如果我从中除去x和t之外的所有参数：

def SIR(x, t):
    return x;

如您所见，导致错误的行是

x = integrate.odeint(SIR, x0, t, args=(beta, gamma, mu, M));

编辑：

我被要求添加SIR方法的完整代码。因为它比较长，所以我将完整的.py脚本放到了一个pastebin中： http://pastebin.com/RphJbCHN

再次感谢。

Answer 1

从tutorial看来，integrate.odeint()的第一个参数需要是一个函数来操作（在您的例子中）x0和t。由于SIR()函数只接受一个参数，因此操作失败。从SIR()返回的结果的大小和/或形状可能与其余参数有关。