python waterfall plot fft