python - 在python中找到切比雪夫多项式的根

Question

我想使用 Python 找到任意阶 Chebysev 多项式的根。我见过类似的勒让德多项式线程。但是，我使用此处定义的方法构造了多项式

import numpy as np 
import sympy as sp 

f0 = lambda x: chebyt(0,x)
f1 = lambda x: chebyt(1,x)
f2 = lambda x: chebyt(2,x)
f3 = lambda x: chebyt(3,x)
f4 = lambda x: chebyt(4,x)
plot([f0,f1,f2,f3,f4],[-1,1])

我尝试使用np.roots(f4)，但收到以下错误：TypeError: float() argument must be a string or a number, not 'function'。此外，似乎即使我可以，它也不适用于高阶多项式。

Answer 1

您可以使用此处“基本评估”标题下的方法找到切比雪夫多项式的系数，然后np.roots在反向列表中使用来生成多项式的根。

使用np.roots(f4)不起作用，因为该roots函数仅接受多项式系数列表，而不是 lambda 函数。

代码：

from mpmath import chebyt, chop, taylor
import numpy as np

for n in range(5):
    print(np.roots(chop(taylor(lambda x: chebyt(n, x), 0, n))[::-1]))

输出：

[]
[0.]
[ 0.70710678 -0.70710678]
[ 0.8660254 -0.8660254  0.       ]
[-0.92387953  0.92387953 -0.38268343  0.38268343]

希望有帮助。