管理包含正整数的表达式的情况

如何解决管理包含正整数的表达式的情况

我有一堆不同的表达方式，我想用 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 ∊ ℤ}