从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。