无法将多个参数传递给 odeint 求解器

Question

我正在尝试对以下等式进行数值积分，其中 omega(3x1) 向量和 theta 是 (4X1) 向量，W(w) 是 4x4 斜对称矩阵

def skew_omega(w):
    """skew symmetric omega (4x4)"""
        print(w)
        return np.array([[0, -w[0], -w[1], -w[2]],
                         [w[0], 0, w[2], -w[1]],
                         [w[1], -w[2], 0, w[0]],
                         [w[2], w[1], -w[0], 0]])


def theta_model(euler_param, w, t):
"""Ode for theta"""
    print(w)
    theta = 0.5*np.matmul(skew_omega(w),euler_param)
    
    return theta

#initial conditions
t = np.array([0, 1])
theta0 = np.array([1, 0, 0, 0])
omega0 = np.array([0.1, 10, 0])

#solve ode    
theta = odeint(theta_model, theta0, t, args = (omega0,))

0.0

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-72-1324ec16154f> in <module>
----> 1 theta = odeint(theta_model, theta0, t, args = (om,))

W:\Anaconda\lib\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, tfirst)
    243                              full_output, rtol, atol, tcrit, h0, hmax, hmin,
    244                              ixpr, mxstep, mxhnil, mxordn, mxords,
--> 245                              int(bool(tfirst)))
    246     if output[-1] < 0:
    247         warning_msg = _msgs[output[-1]] + " Run with full_output = 1 to get quantitative information."

<ipython-input-55-c581c2c300b8> in theta_model(euler_param, w, t)
      1 def theta_model(euler_param, w, t):
      2     print(w)
----> 3     theta = 0.5*np.matmul(skew_omega(w),euler_param)
      4 
      5     return theta

<ipython-input-49-bb06a7f0e258> in skew_omega(w)
      1 def skew_omega(w):
      2     print(w)
----> 3     return np.array([[0, -w[0], -w[1], -w[2]],
      4                      [w[0], 0, w[2], -w[1]],
      5                      [w[1], -w[2], 0, w[0]],

TypeError: 'float' object is not subscriptable

打印的零（（函数中的打印语句）表明 odeint 求解器没有接受初始值 omega0。相反，它给出一个 0.0 浮点数作为占位符，因此抛出 TypeError。

为什么 ode 求解器不将 omega0 作为输入？

Answer 1

来自 odeint 文档

Parameters
----------
func : callable(y, t, ...) or callable(t, y, ...)
Computes the derivative of y at t.

你的模型函数

def theta_model(euler_param, w, t):

看来您必须在参数列表中交换 t 和 w。

无法将多个参数传递给 odeint 求解器

Unable to pass multiple arguments to the odeint solver

python

rigid-bodies

quaternions

scipy

odeint