用matplotlib specgram绘图？

Question

我试图使用matplotlib绘制信号和信号的频谱图，但是......我只得到信号的第一个值（样本）的光谱图（就像30000中的60个... ）。

这是一个非常长的文件，这就是为什么我只想绘制第一个30000样本的原因。

这是代码：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

Data=pd.read_csv('MySignal.txt', 
    skiprows=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19], 
    header=0)
print(Data.head())
DataI=Data['Sig'].tolist()
print(len(Data.index))
DataI=DataI[0:30000]

NFFT = 200     # the length of the windowing segments
Fs = 500  # the sampling rate

# plot signal and spectrogram

t=range(len(DataI))
ax1 = plt.subplot(211)
plt.plot(t, DataI)
plt.subplot(212, sharex=ax1)
Pxx, freqs, bins, im = plt.specgram(DataI, NFFT=NFFT, 
                        Fs=Fs,noverlap=100, cmap=plt.cm.gist_heat)
plt.show() 

我不太了解plt.specgram是如何工作的，所以我不明白问题出在哪里......

非常感谢！

Answer 1

这是一个快速而肮脏的合成示例，三个音调相隔八度，但效果很好。请阅读采样定理以了解频谱图的概念。实际上，最好首先学习如何通过仅使用FFT来绘制光谱（光谱图的垂直切片）。

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

time1 = np.arange(0,5,0.0001)
time = np.arange(0,15,0.0001)
data1=np.sin(2*np.pi*300*time1)
data2=np.sin(2*np.pi*600*time1)
data3=np.sin(2*np.pi*900*time1)
data=np.append(data1,data2 )
data=np.append(data,data3)
print len(time)
print len(data)

NFFT = 200     # the length of the windowing segments
Fs = 500  # the sampling rate

# plot signal and spectrogram

ax1 = plt.subplot(211)
plt.plot(time,data)   # for this one has to either undersample or zoom in 
plt.xlim([0,15])
plt.subplot(212 )  # don't share the axis
Pxx, freqs, bins, im = plt.specgram(data, NFFT=NFFT,   Fs=Fs,noverlap=100, cmap=plt.cm.gist_heat)
plt.show() 

顶部x轴以秒为单位。为了清晰起见，我在5秒的时间内以300 Hz到600 Hz的过渡放大。底轴不是几秒钟，这就是我取出轴共享的原因。这可以修复（细节）。