在scipy中重现sox谱图

import numpy as np
import scipy.io.wavfile as wav
import scipy.signal as signal
import matplotlib.pyplot as plt

# 加载音频文件
sample_rate, audio_data = wav.read('audio.wav')

# 计算频谱
frequencies, times, spectrogram = signal.spectrogram(audio_data, sample_rate)

# 绘制频谱图
plt.pcolormesh(times, frequencies, 10 * np.log10(spectrogram))
plt.xlabel('Time [s]')
plt.ylabel('Frequency [Hz]')
plt.colorbar(label='Power Spectral Density [dB]')
plt.show()