scipy.signal.ShortTimeFFT.
范围#
- ShortTimeFFT.extent(n, axes_seq='tf', center_bins=False)[源代码]#
返回时间-频率值的最小值和最大值。
对于 'tf' 返回一个包含四个浮点数的元组
(t0, t1, f0, f1)
,对于 'ft' 返回(f0, f1, t0, t1)
,描述stft
的时频域的四个角。该元组可以作为同名参数传递给matplotlib.pyplot.imshow
。- 参数:
- nint
输入信号中的采样数。
- axes_seq{‘tf’, ‘ft’}
先返回时间范围,然后返回频率范围,反之亦然。
- center_bins: bool
如果设置(默认
False
),则时间槽和频率箱的值将从侧面移动到中间。当将stft
值绘制为阶跃函数时(即,没有插值时),这很有用。
另请参阅
matplotlib.pyplot.imshow
将数据作为图像显示。
scipy.signal.ShortTimeFFT
此方法所属的类。
示例
以下两个图说明了参数 center_bins 的效果:网格线表示 STFT 的三个时间和四个频率值。左图,其中
(t0, t1, f0, f1) = (0, 3, 0, 4)
作为参数extent
传递给imshow
,显示了时间和频率值的标准行为,即位于相应箱的下边缘。右图,其中(t0, t1, f0, f1) = (-0.5, 2.5, -0.5, 3.5)
,显示当传递center_bins=True
时,箱会居中于各自的值。>>> import matplotlib.pyplot as plt >>> import numpy as np >>> from scipy.signal import ShortTimeFFT ... >>> n, m = 12, 6 >>> SFT = ShortTimeFFT.from_window('hann', fs=m, nperseg=m, noverlap=0) >>> Sxx = SFT.stft(np.cos(np.arange(n))) # produces a colorful plot ... >>> fig, axx = plt.subplots(1, 2, tight_layout=True, figsize=(6., 4.)) >>> for ax_, center_bins in zip(axx, (False, True)): ... ax_.imshow(abs(Sxx), origin='lower', interpolation=None, aspect='equal', ... cmap='viridis', extent=SFT.extent(n, 'tf', center_bins)) ... ax_.set_title(f"{center_bins=}") ... ax_.set_xlabel(f"Time ({SFT.p_num(n)} points, Δt={SFT.delta_t})") ... ax_.set_ylabel(f"Frequency ({SFT.f_pts} points, Δf={SFT.delta_f})") ... ax_.set_xticks(SFT.t(n)) # vertical grid line are timestamps ... ax_.set_yticks(SFT.f) # horizontal grid line are frequency values ... ax_.grid(True) >>> plt.show()
请注意,带有恒定颜色的阶梯状行为是由将
interpolation=None
传递给imshow
引起的。