Python 2d Fourier Transform Example