Question

你知道如何从FFT中删除这么多噪音吗？这是我的FFT代码：

import numpy as np

fft1 = (Bx[51:-14])
fft2 = (By[1:-14])


# Loop for FFT data
for dataset in [fft1]:
    dataset = np.asarray(dataset)
    psd = np.abs(np.fft.fft(dataset))**2
    freq = np.fft.fftfreq(dataset.size, float(300)/dataset.size)
    plt.semilogy(freq[freq>0], psd[freq>0]/dataset.size**2, color='r')


for dataset2 in [fft2]:
    dataset2 = np.asarray(dataset2)
    psd2 = np.abs(np.fft.fft(dataset2))**2
    freq2 = np.fft.fftfreq(dataset2.size, float(300)/dataset2.size)
    plt.semilogy(freq2[freq2>0], psd2[freq2>0]/dataset2.size**2, color='b')

我得到了什么：

我需要什么：

有什么想法吗？韦尔奇不起作用，所以你可以看到，我不想让我的图表变得平滑，而是将这么多的噪音抹去第二张图片上的水平。

这就是韦尔奇所做的：和一些代码：

freqs, psd = scipy.signal.welch(dataset, fs=300, window='hamming')

更新了Welch：

一些代码：

# Loop for FFT data
for dataset in [fft1]:
    dataset = np.asarray(dataset)
    freqs, psd = welch(dataset, fs=266336/300, window='hamming', nperseg=512)
    plt.semilogy(freqs, psd/dataset.size**2, color='r')

for dataset2 in [fft2]:
    dataset2 = np.asarray(dataset2)
    freqs2, psd2 = welch(dataset2, fs=266336/300, window='hamming', nperseg=512)
    plt.semilogy(freqs2, psd2/dataset2.size**2, color='b')

正如您所看到的，Welch配置良好，它显示60 Hz电力线和谐波模式。这几乎是好的，但它完全平滑了我的情节。见图2，这是所需的。顺便说一句。韦尔奇图中的尺度是错误的，但这只是两个电力数据的情况。

我已经改为nperseg = 8192并且它有效。看看结果。

Answer 1

以下是一个示例，说明如何使用nperseg来控制频率分辨率与降噪权衡：

将nperseg设置为信号的长度或多或少等于使用FFT而不进行任何平均。

以下是生成此图片的代码：

import numpy as np
from scipy import signal
import matplotlib.pyplot as plt

plt.figure(figsize=[8, 12])

n = 2**21
fs = 887

# example data
x = np.random.randn(n)
x += np.sin(np.cumsum(0.42 + np.random.randn(n) * 0.01)) * 5
x = signal.lfilter([1, 0.5], 2, x)

plt.subplot(3, 2, 1)
plt.semilogy(np.abs(np.fft.fft(x)[:n//2])**2 / n**2, label='FFT')
plt.legend(loc='best')

for i, nperseg in enumerate([128, 512, 8192, 65536, n]):
    plt.subplot(3, 2, i+2)
    f, psd = signal.welch(x, fs=fs, window='hamming', nperseg=nperseg, noverlap=0)
    plt.semilogy(f, psd, label='nperseg={}'.format(nperseg))
    plt.legend(loc='best')

plt.show()

擦除fft图表中的噪音

1 个答案: