scipy.signal.

spectrogram#

scipy.signal.spectrogram(x, fs=1.0, window=('tukey', 0.25), nperseg=None, noverlap=None, nfft=None, detrend='constant', return_onesided=True, scaling='density', axis=-1, mode='psd')[source]#

使用连续傅里叶变换计算频谱图(遗留函数)。

频谱图可用于可视化非平稳信号频率内容随时间的变化。

遗留功能

此函数被视为遗留功能,将不再接收更新。虽然我们目前没有移除它的计划,但建议新代码改用更现代的替代方案。ShortTimeFFT 是一个更新的 STFT / ISTFT 实现,具有更多功能,也包括一个 spectrogram 方法。有关这些实现之间的比较,请参阅 SciPy 用户指南短时傅里叶变换部分。

参数:
xarray_like

测量值的时间序列

fsfloat, optional

x 时间序列的采样频率。默认为 1.0。

windowstr or tuple or array_like, optional

要使用的所需窗函数。如果 window 是字符串或元组,它将传递给 get_window 以生成窗函数值,这些值默认是 DFT-偶数的。有关窗函数列表和所需参数,请参阅 get_window。如果 window 是 array_like,它将直接用作窗函数,并且其长度必须为 nperseg。默认为形状参数为 0.25 的 Tukey 窗函数。

npersegint, optional

每个段的长度。默认为 None,但如果 window 是 str 或 tuple,则设置为 256;如果 window 是 array_like,则设置为窗函数的长度。

noverlapint, optional

段之间重叠的点数。如果为 None,则 noverlap = nperseg // 8。默认为 None

nfftint, optional

使用的 FFT 长度,如果需要零填充 FFT。如果为 None,则 FFT 长度为 nperseg。默认为 None

detrendstr or function or False, optional

指定如何对每个段进行去趋势。如果 detrend 是字符串,则作为 type 参数传递给 detrend 函数。如果它是一个函数,它接受一个段并返回一个去趋势的段。如果 detrendFalse,则不进行去趋势。默认为 ‘constant’。

return_onesidedbool, optional

如果为 True,则返回实数数据的一侧频谱。如果为 False,则返回双侧频谱。默认为 True,但对于复数数据,总是返回双侧频谱。

scaling{ ‘density’, ‘spectrum’ }, optional

选择计算功率谱密度(‘density’),此时 Sxx 的单位为 V**2/Hz;或计算功率谱(‘spectrum’),此时 Sxx 的单位为 V**2(如果 x 以 V 为单位测量,fs 以 Hz 为单位测量)。默认为 ‘density’。

axisint, optional

计算频谱图的轴;默认为最后一个轴(即 axis=-1)。

modestr, optional

定义预期返回值的类型。选项有 [‘psd’, ‘complex’, ‘magnitude’, ‘angle’, ‘phase’]。“complex”等同于不进行填充或边界扩展的 stft 输出。“magnitude”返回 STFT 的绝对幅值。“angle”和“phase”分别返回 STFT 的复数角度(带或不带解卷绕)。

返回值:
fndarray

采样频率数组。

tndarray

段时间数组。

Sxxndarray

x 的频谱图。默认情况下,Sxx 的最后一个轴对应于段时间。

另请参阅

periodogram

简单、可选修改的周期图

lombscargle

用于不均匀采样数据的 Lomb-Scargle 周期图

welch

基于 Welch 方法的功率谱密度。

csd

基于 Welch 方法的互谱密度。

ShortTimeFFT

提供更多功能的新型 STFT/ISTFT 实现,其中也包括一个 spectrogram 方法。

注释

适当的重叠量将取决于窗函数的选择和您的要求。与 Welch 方法(对整个数据流进行平均)不同,在计算频谱图时,可能希望使用较小的重叠(或完全不重叠),以保持各个段之间的一些统计独立性。正是出于这个原因,默认的窗函数是一个 Tukey 窗,其两端各重叠窗函数长度的 1/8。

在 0.16.0 版本中添加。

参考文献

[1]

Oppenheim, Alan V., Ronald W. Schafer, John R. Buck “Discrete-Time Signal Processing”, Prentice Hall, 1999.

示例

>>> import numpy as np
>>> from scipy import signal
>>> from scipy.fft import fftshift
>>> import matplotlib.pyplot as plt
>>> rng = np.random.default_rng()

生成一个测试信号:一个 2 Vrms 正弦波,其频率在 3kHz 附近缓慢调制,并受到幅度呈指数衰减的白噪声干扰,采样频率为 10 kHz。

>>> fs = 10e3
>>> N = 1e5
>>> amp = 2 * np.sqrt(2)
>>> noise_power = 0.01 * fs / 2
>>> time = np.arange(N) / float(fs)
>>> mod = 500*np.cos(2*np.pi*0.25*time)
>>> carrier = amp * np.sin(2*np.pi*3e3*time + mod)
>>> noise = rng.normal(scale=np.sqrt(noise_power), size=time.shape)
>>> noise *= np.exp(-time/5)
>>> x = carrier + noise

计算并绘制频谱图。

>>> f, t, Sxx = signal.spectrogram(x, fs)
>>> plt.pcolormesh(t, f, Sxx, shading='gouraud')
>>> plt.ylabel('Frequency [Hz]')
>>> plt.xlabel('Time [sec]')
>>> plt.show()
../../_images/scipy-signal-spectrogram-1_00_00.png

注意,如果使用非单侧输出,请使用以下方法

>>> f, t, Sxx = signal.spectrogram(x, fs, return_onesided=False)
>>> plt.pcolormesh(t, fftshift(f), fftshift(Sxx, axes=0), shading='gouraud')
>>> plt.ylabel('Frequency [Hz]')
>>> plt.xlabel('Time [sec]')
>>> plt.show()
../../_images/scipy-signal-spectrogram-1_01_00.png