Следующий код создает спектрограмму, используя либо scipy.signal.spectrogram
, либо matplotlib.pyplot.specgram
.
Однако цветовой контраст функции specgram
довольно низкий. Есть ли способ увеличить его?
import numpy as np
from scipy import signal
import matplotlib.pyplot as plt
# Generate data
fs = 10e3
N = 5e4
amp = 4 * np.sqrt(2)
noise_power = 0.01 * fs / 2
time = np.arange(N) / float(fs)
mod = 800*np.cos(2*np.pi*0.2*time)
carrier = amp * np.sin(2*np.pi*time + mod)
noise = np.random.normal(scale=np.sqrt(noise_power), size=time.shape)
noise *= np.exp(-time/5)
x = carrier + noise
Использование matplotlib.pyplot.specgram
дает следующий результат:
Pxx, freqs, bins, im = plt.specgram(x, NFFT=1028, Fs=fs)
x1, x2, y1, y2 = plt.axis()
plt.axis((x1, x2, 0, 200))
plt.show()
Использование scipy.signal.spectrogram
дает следующий график
f, t, Sxx = signal.spectrogram(x, fs, nfft=1028)
plt.pcolormesh(t, f[0:20], Sxx[0:20])
plt.ylabel('Frequency [Hz]')
plt.xlabel('Time [sec]')
plt.show()
Обе функции, кажется, используют цветовую карту "jet".
Меня также вообще интересовала бы разница между двумя функциями. Хотя они делают что-то похожее, они явно не идентичны.
matplotlib.pyplot.specgram
содержит в основном теплые цвета (желтый) на заднем плане, тогда как графикscipy.signal.spectrogram contains rather cold colors (blue) in the background. I would like to achieve scipy's choice of colors for matplotlib's
specgram`. - person NicolasBourbaki   schedule 04.02.2018