Python 2d Fourier Transform