Short Time Fourier Transform Python