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

Question

我有一堆不同的表达我想用 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

标签： pythonsympysimplify

Answer 1

我仍然不清楚你真正想要什么，但也许这会有所帮助。如果您不声明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 ∊ ℤ}