Spectral methods in python