符号表达式中变量的隔离

Question

我有一个把符号拉格朗日方程转换成哈密顿方程的函数。我遇到的一个问题是用动量项代替质量*速度项

我有一行代码返回所有动量项：

momentum = lagrangian.jacobian(velocity)

返回一个以质量*速度为单位的符号矩阵 示例：

>>> momentum
Matrix([[0, 0, 1.0*m_1*xd2, 1.0*m_1*yd2, 1.0*m_2*td1, 1.0*m_2*td2]])

当我试着用动量符号“p”来代替每一项时，sympy.subs函数并不能识别任何质量*速度项。我相信这是因为表达式中的常量1.0。我很好奇是否有办法隔离符号项或者去掉符号项前面的常量

Answer 1

当你说“去掉常数”时，我想你的意思是你想让1.0消失。一种方法是在矩阵上使用nsimplify：

>>> m
Matrix([[1.0*m*v**2]])
>>> nsimplify(m, rational=True)
Matrix([[m*v**2]])