Python中的系数矩阵 - 组元素

时间:2014-06-25 22:53:16

标签: python matrix sympy coefficients

我使用Sympy在Python中找到了这个矩阵:

> Matrix([[-THETA*l*m2*omega**2*cos(omega*t) + X*k*cos(omega*t) -
> X*omega**2*(m1 + m2)*cos(omega*t)], [THETA*g*cos(omega*t) -
> THETA*l*omega**2*cos(omega*t) - X*omega**2*cos(omega*t)]])

我现在需要找到一个表达式:

[coefficient matrix]*(unknowns vectors)

我的(未知数矢量)是:

Matrix([[X],[THETA]]).

我试图使用Sympy解决,简化和收集而没有成功(我只能得到错误或[]返回)。

1 个答案:

答案 0 :(得分:1)

选择Jacobian

In [16]: a.jacobian(Matrix([X, THETA]))
Out[16]:
⎡              2                                2              ⎤
⎢k⋅cos(ω⋅t) - ω ⋅(m₁ + m₂)⋅cos(ω⋅t)      -l⋅m₂⋅ω ⋅cos(ω⋅t)     ⎥
⎢                                                              ⎥
⎢             2                                      2         ⎥
⎣           -ω ⋅cos(ω⋅t)             g⋅cos(ω⋅t) - l⋅ω ⋅cos(ω⋅t)⎦

In [17]: a.jacobian(Matrix([X, THETA]))*Matrix([X, THETA])
Out[17]:
⎡              2              ⎛              2                   ⎞⎤
⎢- THETA⋅l⋅m₂⋅ω ⋅cos(ω⋅t) + X⋅⎝k⋅cos(ω⋅t) - ω ⋅(m₁ + m₂)⋅cos(ω⋅t)⎠⎥
⎢                                                                 ⎥
⎢             ⎛                2         ⎞      2                 ⎥
⎣       THETA⋅⎝g⋅cos(ω⋅t) - l⋅ω ⋅cos(ω⋅t)⎠ - X⋅ω ⋅cos(ω⋅t)        ⎦

In [22]: a
Out[22]:
⎡              2                              2                   ⎤
⎢- THETA⋅l⋅m₂⋅ω ⋅cos(ω⋅t) + X⋅k⋅cos(ω⋅t) - X⋅ω ⋅(m₁ + m₂)⋅cos(ω⋅t)⎥
⎢                                                                 ⎥
⎢                                 2               2               ⎥
⎣     THETA⋅g⋅cos(ω⋅t) - THETA⋅l⋅ω ⋅cos(ω⋅t) - X⋅ω ⋅cos(ω⋅t)      ⎦

顺便说一句,如果您使用thetaTheta(并非全部大写),SymPy会将其打印为实际的希腊字母theta:

In [24]: symbols('theta')
Out[24]: θ