本文作者: slience_me
文章目录
- 关于去除信号中的直流分量效果演示(零频率分量)
- 1. 效果图展示:
- 2. 快速傅里叶变换FFT
- 3. 相关验证代码
关于去除信号中的直流分量效果演示(零频率分量)
1. 效果图展示:
2. 快速傅里叶变换FFT
def FFT_for_Period(x, k=2):xf = torch.fft.rfft(x, dim=1)frequency_list = abs(xf).mean(0).mean(-1)frequency_list[0] = 0_, top_list = torch.topk(frequency_list, k)top_list = top_list.detach().cpu().numpy()period = x.shape[1] // top_listreturn period, abs(xf).mean(-1)[:, top_list]
将 frequency_list[0]
置零可能是为了去除直流分量(零频率分量)的影响。在频率谱中,索引为0的位置通常对应于零频率,也就是直流成分。
直流成分表示信号的平均值或偏移。在某些情况下,我们可能对信号的变化更感兴趣,而不是整个信号的平均值。通过将 frequency_list[0]
置零,我们可以去除直流成分的影响,更关注信号中的变化和其他频率成分。
这种操作在信号处理和频谱分析中是常见的,特别是当我们关注信号的变化或周期性成分而不关心平均水平时。它有助于突出频谱中的其他特征,使得分析更加集中于信号的变动和周期性。
关于直流分量的解释
直流分量是信号中的恒定成分,通常表示信号的平均值或直流偏移。直流分量不随时间变化,它是信号在水平方向上的偏移或平移。
让我们通过一个直观的例子来理解直流分量:
假设有一个以时间为横轴的信号图,其中纵轴表示信号的振幅。如果信号在整个时间范围内都有一个常数振幅,那么这个信号就包含直流分量。直流分量的存在会使整个信号在纵轴上发生平移,即整个信号的基准水平线上下移动。
举个例子,考虑一个表示温度的信号。如果这个信号中存在直流分量,那么它可能表示一个常数的环境温度,而信号的波动则表示温度随时间的变化。直流分量可以看作是整个信号的平均温度,而波动则反映了温度相对于平均值的变化。
在频谱分析中,直流分量通常对应于频谱中的零频率分量,即索引为0的位置。通过将直流分量从频谱中去除,我们可以更专注于信号中变化的频率成分,而不受整体平移的影响。这在很多信号处理任务中是有用的,特别是当我们关注信号的变动和周期性成分时。
3. 相关验证代码
import numpy as np
import matplotlib.pyplot as plt
import torchdef generate_irregular_waveform_with_large_dc(duration, sampling_rate):t = np.arange(0, duration, 1 / sampling_rate)# 生成不规则的波形,包含直流分量waveform = np.sin(2 * np.pi * 5 * t) + 0.5 * np.sin(2 * np.pi * 12 * t) + 2.0 # 增大直流分量return t, waveform# 设置不规则波形的参数
duration = 2 # 波形持续时间(秒)
sampling_rate = 1000 # 采样率# 生成不规则波形数据(包含直流分量)
t, waveform_with_dc = generate_irregular_waveform_with_large_dc(duration, sampling_rate)# 转换为PyTorch张量
x_with_dc = torch.tensor(waveform_with_dc, dtype=torch.float32)# 不去掉直流分量的傅里叶变换
xf_with_dc = torch.fft.rfft(x_with_dc)
print("不去掉直流分量")
print("xf.shape: {}".format(xf_with_dc.shape))
frequency_list = abs(xf_with_dc)
print("frequency_list: {}".format(frequency_list))
_, top_list = torch.topk(frequency_list, 2)
print("top_list: {}".format(top_list))# 提取频率信息
xf_freq = np.fft.fftfreq(len(waveform_with_dc), d=1 / sampling_rate)plt.figure(figsize=(24, 16))
# 绘制原始波形图
plt.subplot(4, 1, 1)
plt.plot(t, waveform_with_dc)
plt.title('Original Waveform (with DC Component)')
plt.xlabel('Time (s)')
plt.ylabel('Amplitude')
plt.grid(True)# 绘制不去掉直流分量的频谱图
plt.subplot(4, 1, 2)
plt.plot(xf_freq[:len(xf_with_dc)], torch.abs(xf_with_dc))
plt.title('Frequency Spectrum (without Removing DC Component)')
plt.xlabel('Frequency (Hz)')
plt.ylabel('Amplitude')
plt.grid(True)# 去掉直流分量
x_without_dc = x_with_dc - torch.mean(x_with_dc)
print("去掉直流分量")
print("xf.shape: {}".format(xf_with_dc.shape))
frequency_list = abs(xf_with_dc)
print("frequency_list: {}".format(frequency_list))
_, top_list = torch.topk(frequency_list, 2)
print("top_list: {}".format(top_list))# 进行傅里叶变换
xf_without_dc = torch.fft.rfft(x_without_dc)# 提取频率信息
xf_freq_without_dc = np.fft.fftfreq(len(waveform_with_dc), d=1 / sampling_rate)# 绘制去掉直流分量后的波形图
plt.subplot(4, 1, 3)
plt.plot(t, x_without_dc.numpy())
plt.title('Waveform without DC Component')
plt.xlabel('Time (s)')
plt.ylabel('Amplitude')
plt.grid(True)# 绘制去掉直流分量后的频谱图
plt.subplot(4, 1, 4)
plt.plot(xf_freq_without_dc[:len(xf_without_dc)], torch.abs(xf_without_dc))
plt.title('Frequency Spectrum')
plt.xlabel('Frequency (Hz)')
plt.ylabel('Amplitude')
plt.grid(True)plt.tight_layout()
plt.show()