Fourier transform python example