diff --git a/pina/chebyshev.py b/pina/chebyshev.py index 6a9fa15..89cb1e6 100644 --- a/pina/chebyshev.py +++ b/pina/chebyshev.py @@ -1,6 +1,7 @@ -from mpmath import chebyt, chop, taylor import numpy as np def chebyshev_roots(n): """ Return the roots of *n* Chebyshev polynomials (between [-1, 1]) """ - return np.roots(chop(taylor(lambda x: chebyt(n, x), 0, n))[::-1]) + coefficents = np.zeros(n+1) + coefficents[-1] = 1 + return np.polynomial.chebyshev.chebroots(coefficents)