Python jupyter notebook fourier transform