wavshow.py

import numpy as np
import matplotlib.pyplot as plt

def wavshow(audio, sr=None, resolution=2048):
    duration = len(audio)
    if sr is not None:
         duration /= sr
    plt.figure(figsize=(16,4))
    n = len(audio)
    k = n // resolution
    if k > 1:
        audio = audio[:(n//k)*k]
        audio = audio.reshape(-1,k)
        mean = audio.mean(axis=1)
        std = audio.std(axis=1)
        x_ticks = np.linspace(0, duration, audio.shape[0])
        plt.fill_between(x_ticks, mean - std, mean + std, color='black', lw=0)
        plt.fill_between(x_ticks, audio.min(axis=1), audio.max(axis=1), alpha=0.5, color='gray', lw=0)
    else:
        x_ticks = np.linspace(0, duration, len(audio))
        plt.plot(x_ticks, audio, color='black', lw=1)
    plt.xlim(0, duration)
    plt.xlabel('Seconds' if sr is not None else 'Samples')
    plt.show()