如何使用SymPy替换模式

时间:2016-10-23 19:04:12

标签: python sympy symbolic-math

在我的代码中,我需要将sin(g(t))g是连续函数)的所有表达式替换为g(t)(这是紧密角度近似)。这是我从代码中得到的一个示例:

-29.4*sin(2*t) - 19.6*sin(f(t)) + 4.0*Derivative(f(t), t)**2

我需要替换sin(f(t))sin(2*t)。不只是其中一个和sin(2*t)更改,(sin(f(t))始终相同)。有没有比为sin内部添加额外变量更简单的方法,还是不存在?

1 个答案:

答案 0 :(得分:5)

这是你想要做的吗?

import sympy as sp

t = sp.symbols('t')
f = sp.Function('f')

expr_v1 = -29.4*sp.sin(2*t) - 19.6*sp.sin(f(t)) + 4.0*sp.Derivative(f(t), t)**2
print('expr_v1 = ', expr_v1)

expr_v2 = expr_v1.replace(sp.sin, lambda *args: args[0])
print('expr_v2 = ', expr_v2)
expr_v1 =  -29.4*sin(2*t) - 19.6*sin(f(t)) + 4.0*Derivative(f(t),t)**2
expr_v2 =  -58.8*t - 19.6*f(t) + 4.0*Derivative(f(t), t)**2