scipy.signal.

stft#

scipy.signal.stft(x, fs=1.0, window='hann', nperseg=256, noverlap=None, nfft=None, detrend=False, return_onesided=True, boundary='zeros', padded=True, axis=-1, scaling='spectrum')[源代码]#

计算短时傅里叶变换（遗留函数）。

STFT 可用于量化非平稳信号的频率和相位内容随时间的变化。

遗留

此函数被认为是遗留函数，将不再接收更新。虽然我们目前没有计划删除它，但我们建议新代码使用更现代的替代方案。ShortTimeFFT 是一个更新的 STFT / ISTFT 实现，具有更多功能。在比较实现之间，可以在短时傅里叶变换部分中找到SciPy 用户指南。

参数:

xarray_like: 测量值的时间序列
fsfloat，可选: x时间序列的采样频率。默认为 1.0。
windowstr 或 tuple 或 array_like，可选: 要使用的所需窗口。如果 window 是字符串或元组，则将其传递给 get_window 以生成窗口值，默认情况下窗口值为 DFT 偶数。有关窗口和所需参数的列表，请参见 get_window。如果 window 是 array_like，则它将直接用作窗口，并且其长度必须为 nperseg。默认为 Hann 窗口。
npersegint，可选: 每个分段的长度。默认为 256。
noverlapint，可选: 分段之间重叠的点数。如果 None，noverlap = nperseg // 2。默认为 None。指定时，必须满足 COLA 约束（请参阅下面的注释）。
nfftint，可选: 如果需要零填充 FFT，则使用的 FFT 长度。如果 None，则 FFT 长度为 nperseg。默认为 None。
detrendstr 或 function 或 False，可选: 指定如何对每个分段进行去趋势。如果 detrend 是字符串，则将其作为 type 参数传递给 detrend 函数。如果它是函数，则它接受一个分段并返回一个去趋势的分段。如果 detrend 是 False，则不进行去趋势。默认为 False。
return_onesidedbool，可选: 如果 True，则返回实数数据的一侧频谱。如果 False，则返回双边频谱。默认为 True，但对于复数数据，始终返回双边频谱。
boundarystr 或 None，可选: 指定输入信号是否在两端扩展，以及如何生成新值，以便将第一个加窗分段置于第一个输入点上。这样做的好处是，当所采用的窗口函数从零开始时，可以重建第一个输入点。有效选项为 ['even', 'odd', 'constant', 'zeros', None]。默认为 ‘zeros’，用于零填充扩展。即，对于 nperseg=3，[1, 2, 3, 4] 会被扩展为 [0, 1, 2, 3, 4, 0]。
paddedbool，可选: 指定是否在末尾对输入信号进行零填充，以使信号完全适合整数个窗口分段，从而使所有信号都包含在输出中。默认为 True。如果 boundary 不为 None，并且 padded 为 True（默认值），则会在边界扩展之后进行填充。
axisint，可选: 计算 STFT 沿着的轴；默认值是沿最后一个轴（即 axis=-1）。
scaling: {‘spectrum’, ‘psd’}: 默认的 ‘spectrum’ 缩放允许将 Zxx 的每条频率线解释为幅度频谱。’psd’ 选项将每条线缩放到功率谱密度 - 它允许通过对 abs(Zxx)**2 进行数值积分来计算信号的能量。

1.9.0 版本中新增。

返回:

fndarray: 采样频率数组。
tndarray: 分段时间数组。
Zxxndarray: x 的 STFT。默认情况下，Zxx 的最后一个轴对应于分段时间。

另请参阅

istft: 逆短时傅里叶变换
ShortTimeFFT: 提供更多功能的较新 STFT/ISTFT 实现。
check_COLA: 检查是否满足恒定重叠相加 (COLA) 约束
check_NOLA: 检查是否满足非零重叠相加 (NOLA) 约束
welch: 通过 Welch 方法计算功率谱密度。
spectrogram: 通过 Welch 方法计算频谱图。
csd: 通过 Welch 方法计算互谱密度。
lombscargle: 用于不均匀采样数据的 Lomb-Scargle 周期图

注释

为了可以通过 istft 中的逆 STFT 实现 STFT 的反演，信号加窗必须遵守“非零重叠相加”（NOLA）的约束，并且输入信号必须具有完整的加窗覆盖率（即 (x.shape[axis] - nperseg) % (nperseg-noverlap) == 0）。可以使用 padded 参数来实现这一点。

给定一个时域信号 \(x[n]\)，一个窗口 \(w[n]\)，和一个跳跃大小 \(H\) = nperseg - noverlap，时间索引为 \(t\) 的加窗帧由下式给出

\[x_{t}[n]=x[n]w[n-tH]\]

重叠相加 (OLA) 重建方程由下式给出

\[x[n]=\frac{\sum_{t}x_{t}[n]w[n-tH]}{\sum_{t}w^{2}[n-tH]}\]

NOLA 约束确保 OLA 重建方程分母中出现的每个归一化项都不为零。可以使用 check_NOLA 测试 window、nperseg 和 noverlap 的选择是否满足此约束。

0.19.0 版本中新增。

参考文献

[1]

Oppenheim, Alan V., Ronald W. Schafer, John R. Buck “离散时间信号处理”，Prentice Hall，1999。

[2]

Daniel W. Griffin, Jae S. Lim “从修改后的短时傅里叶变换进行信号估计”，IEEE 1984, 10.1109/TASSP.1984.1164317

示例

>>> import numpy as np
>>> from scipy import signal
>>> 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

计算并绘制 STFT 的幅度。

>>> f, t, Zxx = signal.stft(x, fs, nperseg=1000)
>>> plt.pcolormesh(t, f, np.abs(Zxx), vmin=0, vmax=amp, shading='gouraud')
>>> plt.title('STFT Magnitude')
>>> plt.ylabel('Frequency [Hz]')
>>> plt.xlabel('Time [sec]')
>>> plt.show()

../../_images/scipy-signal-stft-1_00_00.png

将信号 x 的能量与其 STFT 的能量进行比较

>>> E_x = sum(x**2) / fs  # Energy of x
>>> # Calculate a two-sided STFT with PSD scaling:
>>> f, t, Zxx = signal.stft(x, fs, nperseg=1000, return_onesided=False,
...                         scaling='psd')
>>> # Integrate numerically over abs(Zxx)**2:
>>> df, dt = f[1] - f[0], t[1] - t[0]
>>> E_Zxx = sum(np.sum(Zxx.real**2 + Zxx.imag**2, axis=0) * df) * dt
>>> # The energy is the same, but the numerical errors are quite large:
>>> np.isclose(E_x, E_Zxx, rtol=1e-2)
True