Spectrogram audio python matplotlib: how to change color and add intensity scale?

Question

Python

There is a code

import matplotlib.pyplot as plt
import scipy.io.wavfile
sample_rate, X = scipy.io.wavfile.read('TEST2.wav')
print(sample_rate, X.shape)
plt.specgram(X[:, 0], NFFT=256, Fs=sample_rate)
plt.show()

which gives

I would like to get something like

, i.e., more colorful and with an intensity
scale

Answer 1

Hooray!
Thanks Andy_U !

import matplotlib.pyplot as plt
import scipy.io.wavfile
sample_rate, X = scipy.io.wavfile.read('TEST2.wav')
print(sample_rate, X.shape)
plt.xlabel('Время, c', fontsize='x-large')
plt.ylabel('Частота, Гц', fontsize='x-large')
plt.title('Спектрограмма для колбы')
cmap = plt.get_cmap('magma')
plt.specgram(X[:, 0], NFFT=256, pad_to=256, mode='magnitude', Fs=sample_rate, cmap=cmap)
plt.colorbar()
plt.show()