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")