如何使用python绘制整个音频文件的频谱或频率与振幅的关系？

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

sample_rate, audio_data = wav.read('audio_file.wav')

audio_data = audio_data.mean(axis=1)

fft_data = np.fft.fft(audio_data)

freq_axis = np.fft.fftfreq(len(fft_data), 1/sample_rate)

plt.plot(freq_axis, np.abs(fft_data))
plt.xlabel('Frequency (Hz)')
plt.ylabel('Amplitude')
plt.title('Spectrum of Audio')
plt.show()

pip install numpy matplotlib scipy