스펙트로 그램에 색을 어떻게 추가합니까?

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

specgram 명령 자체에 cmap 인수를 추가 할 수 있습니다. specgram docs을 참조하십시오. Colormaps reference에서 적합한 컬러 맵을 선택할 수 있습니다. 예 명령은 다음과 같습니다 필요한 경우, 당신은 또한 무엇 각 색상 수단을 보여주는 측면에 colorbar을 추가 할 수 있습니다

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

.