Python - 删除范围之间的特定频率

时间:2014-02-03 01:29:30

标签: python audio spectrogram

我有一个频谱图:

enter image description here

我想清理频谱图,所以我只捕获特定范围内的频率(即在本例中,在2627 - 3939之间),并删除低于此频率的所有块。我的总体目标是只保留在此频率范围内的4个段,并且可以识别。

到目前为止,这是我的代码:

import wave, struct, numpy as np, matplotlib.mlab as mlab, pylab as pl
def wavToArr(wavefile):
    w = wave.open(wavefile,"rb")
    p = w.getparams()
    s = w.readframes(p[3])
    w.close()
    sd = np.fromstring(s, np.int16)
    return sd,p

def wavToSpec(wavefile,log=False,norm=False):
    wavArr,wavParams = wavToArr(wavefile)
    print wavParams
    return  mlab.specgram(wavArr, NFFT=256,Fs=wavParams[2],window=mlab.window_hanning,noverlap=128,sides='onesided',scale_by_freq=True)

wavArr,wavParams = wavToArr("4bats.wav")
Pxx, freqs, bins = wavToSpec("4bats.wav")
Pxx += 0.0001

freqs += (len(wavArr) / wavParams[2]) / 2.
hf=pl.figure(figsize=(12,12));
ax = hf.add_subplot(2,1,1);
#plot spectrogram as decibals
hm = ax.imshow(10*np.log10(Pxx),interpolation='nearest',origin='lower',aspect='auto')
hf.colorbar(hm)
ylcnt = len(ax.get_yticklabels())
ycnt = len(freqs)
ylstep = int(ycnt / ylcnt)
ax.set_yticklabels([ int(freqs[f]) for f in xrange(0,ycnt,ylstep) ])
pl.show()

问题是,我不知道如何使用Python来做到这一点。我知道范围(2627 - 3939)但是,我会遍历整个2D阵列并总结所有块,或者,对于频谱图中的每个块,计算频率,如果它高于阈值,保持它,否则值变为0.0?

如果我总结每个箱子,我会得到以下内容:

enter image description here

我需要保留这些块,但是,除了这些块之外,还要删除所有其他块。

我希望有人可以帮助我!

1 个答案:

答案 0 :(得分:1)

也许你想要这样的东西:

Pxx[np.greater(np.sum(Pxx, axis=1), 2955), :] = 0.

或者,我可能会切换你的轴,所以也试试:

Pxx[:, np.greater(np.sum(Pxx, axis=0), 2955)] = 0.

或许你想要别的东西......我觉得这个问题有点不清楚。