Python Discrete Time Fourier Transform