如何在python中绘制wav文件中的波形？

import wave
import numpy as np
import matplotlib.pyplot as plt

with wave.open('example.wav', 'rb') as file:
    # 获取音频参数
    params = file.getparams()
    # 读取音频数据
    frames = file.readframes(params.nframes)
    # 将音频数据转换为一维数组
    audio = np.frombuffer(frames, dtype=np.int16)

# 计算时间轴
time = np.arange(0, params.nframes) / params.framerate

# 绘制波形图
plt.plot(time, audio)
plt.xlabel('Time (s)')
plt.ylabel('Amplitude')
plt.title('Waveform of Audio')
plt.show()