Python Fourier Transform Example Problem