pyz.py

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

def pyz_viz(y, sr):
    plt.figure(figsize=(10, 4))
    chroma = librosa.feature.chroma_stft(y=y, sr=sr)
    librosa.display.specshow(chroma, sr=sr, x_axis='time', y_axis='chroma')
    plt.title("Chromagram")