Question

我试图计算 k ^th 切比雪夫多项式的系数。我们只需将 k 设置为5即可。到目前为止，我有以下内容：

a = (0,0,0,0,0,1) #selects the 5th Chebyshev polynomial
p = numpy.polynomial.chebyshev.Chebyshev(a) #type here is Chebyshev
cpoly = numpy.polynomial.chebyshev.cheb2poly(p) #trying to convert to Poly
print cpoly.all_coeffs()

第二行运行后，我有一个Chebyshev类型的对象，如预期的那样。但是，第三行无法转换为Poly类型，并转换为类型numpy.ndarray。因此，我收到一条错误消息，指出ndarray没有属性all_coeffs。

任何人都知道如何解决这个问题？

Answer 1

@cel 在评论中有正确的想法 - 您需要将Chebyshev多项式的系数传递给cheb2poly，而不是对象本身：< / p>

import numpy as np

cheb = np.polynomial.chebyshev.Chebyshev((0,0,0,0,0,1))
coef = np.polynomial.chebyshev.cheb2poly(cheb.coef)

print(coef)
# [  0.,   5.,   0., -20.,   0.,  16.]

即。 16x ⁵ - 20x ³ + 5x 。您可以确认这些是正确的系数here。

要将这些系数转换为Polynomial对象，只需将数组传递给Polynomial构造函数：

poly = np.polynomial.Polynomial(coef)

Answer 2

In [1]: import numpy.polynomial

In [2]: p = numpy.polynomial.Chebyshev.basis(5)

In [3]: p
Out[3]: Chebyshev([ 0.,  0.,  0.,  0.,  0.,  1.], [-1.,  1.], [-1.,  1.])

In [4]: p.convert(kind=numpy.polynomial.Polynomial)
Out[4]: Polynomial([  0.,   5.,   0., -20.,   0.,  16.], [-1.,  1.], [-1.,  1.])

在Python中生成Chebyshev多项式的系数

2 个答案: