スペクトログラムに色を追加するにはどうすればいいですか？

Question

私はスペクトログラムを吐き出す次のコードを見つけました。しかし、私はカラーマップやその他の方法で色をランダムに追加したいと思っています。私はcmapのドキュメントを読んだことがあります。

コード：あなたの助けを事前に

import matplotlib.pyplot as plt 
from scipy.io import wavfile 

def graph_spectrogram(wav_file): 
    rate, data = get_wav_info(wav_file) 
    nfft = 256 
    fs = 256  
    pxx, freqs, bins, im = plt.specgram(data, nfft,fs) 
    plt.axis('off') 
    plt.savefig('sp_xyz.png', 
       dpi=100, # Dots per inch 
       frameon='false', 
       aspect='normal', 
       bbox_inches='tight', 
       pad_inches=0) 
    plt.show() 

def get_wav_info(wav_file): 
    rate, data = wavfile.read(wav_file) 
    return rate, data 

if __name__ == '__main__': # Main function 
    wav_file = 'song.wav' 
    graph_spectrogram(wav_file) 

ありがとう！

Answer 1

cmap引数をspecgramコマンド自体に追加できます。 specgram docsを参照してください。 Colormaps referenceからあなたに合ったカラーマップを選択することができます。コマンド例は次のようになります。

pxx, freqs, bins, im = plt.specgram(data, nfft, fs, cmap='plasma') 

必要に応じ、あなたはまた、それぞれの色が何を意味するかを示す、側面にcolorbarを追加することができます。