问题描述
我有一堆不同的表达方式,我想用 sympy 来简化。一个非常简单的例子是
form sympy import *
k=Symbol('k',integer=True,nonnegative=True)
term=( 1-exp(-j*2*pi*k) )/(1-exp(-j*2*pi*k/5) )
因此,我得到 term=0,但我希望 k=0 时 term=5,在所有其他情况下,term 为零。有没有一种简单的方法可以简化 k=0 和 k!=0
解决方法
我仍然不清楚您真正想要什么,但也许这会有所帮助。如果您不将 k
声明为整数,则分子不会自动计算为零,然后您可以分别操作分子和分母表达式以找出它们何时为零:
In [24]: k = Symbol('k')
In [25]: term = (1 - exp(-I*2*pi*k))/(1 - exp(-I*2*pi*k/5))
In [26]: num,den = term.as_numer_denom()
In [27]: solveset(num,k)
Out[27]: ℤ
In [28]: solveset(den,k)
Out[28]: {5⋅n │ n ∊ ℤ}