python - 从 mp3 绘制频谱图

Question

我正在尝试直接从 python 2.7.3（使用 ubuntu）中的 mp3 文件绘制频谱图。我可以从 wav 文件中执行以下操作。

#!/usr/bin/python
from scikits.audiolab import wavread
from pylab import *

signal, fs, enc = wavread('XC124158.wav')
specgram(signal)
show()

从 mp3 文件而不是 wav 做同样事情的最干净的方法是什么？如果可以避免的话，我不想将所有 mp3 文件转换为 wav。

Answer 1

我会安装 Debian/Ubuntu 软件包libav-tools并调用avconv将 mp3 解码为临时 wav 文件：

---

编辑：您的另一个问题已关闭，因此我将在这里通过一个简单的带通滤波示例扩展我的答案。在您链接的文件中，看起来大部分鸟鸣都集中在 4 kHz - 5.5 kHz 中。

import os
from subprocess import check_call
from tempfile import mktemp
from scikits.audiolab import wavread, play
from scipy.signal import remez, lfilter
from pylab import *

# convert mp3, read wav
mp3filename = 'XC124158.mp3'
wname = mktemp('.wav')
check_call(['avconv', '-i', mp3filename, wname])
sig, fs, enc = wavread(wname)
os.unlink(wname)

# bandpass filter
bands = array([0,3500,4000,5500,6000,fs/2.0]) / fs
desired = [0, 1, 0]
b = remez(513, bands, desired)
sig_filt = lfilter(b, 1, sig)
sig_filt /=  1.05 * max(abs(sig_filt)) # normalize

subplot(211)
specgram(sig, Fs=fs, NFFT=1024, noverlap=0)
axis('tight'); axis(ymax=8000)
title('Original')
subplot(212)
specgram(sig_filt, Fs=fs, NFFT=1024, noverlap=0)
axis('tight'); axis(ymax=8000)
title('Filtered')
show()

play(sig_filt, fs)

Answer 2

另一种绘制 mp3 文件频谱图的非常简单的方法。

from pydub import AudioSegment
import matplotlib.pyplot as plt
from scipy.io import wavfile
from tempfile import mktemp

mp3_audio = AudioSegment.from_file('speech.mp3', format="mp3")  # read mp3
wname = mktemp('.wav')  # use temporary file
mp3_audio.export(wname, format="wav")  # convert to wav
FS, data = wavfile.read(wname)  # read wav file
plt.specgram(data, Fs=FS, NFFT=128, noverlap=0)  # plot
plt.show()

这使用了pydub比调用外部命令更方便的库。这样，您可以遍历所有.mp3文件，而无需.wav在绘图之前将它们转换为。