In [1]:
%matplotlib inline
from extract_fft import *
import numpy as np
import scipy.io.wavfile as wav
import matplotlib.pyplot as plt
In [2]:
spec, timevec, freqvec = getFFT('test.wav')
print('Spectral data shape: {}'.format(spec.shape))
print('Time vector length: {}'.format(len(timevec)))
print('Frequency vector length: {}'.format(len((freqvec))))
plt.plot(spec[0,:]) # first fft vector
plt.xlim(0, len(freqvec)-1)
tickloc = [int(i) for i in np.linspace(0, len(freqvec)-1, 4)]
plt.xticks(tickloc, [freqvec[t] for t in tickloc])
plt.title('First Spectral Slice')
plt.show()
In [3]:
plotstft("test.wav")