Parseval定理不适用于正弦曲线+噪声的FFT?

时间:2017-10-18 13:35:26

标签: python-2.7 numpy fft

提前感谢您对此主题的任何帮助。我最近一直试图在包含噪声时计算出Parseval定理的离散傅里叶变换。我的代码来自this code

我期望看到的是(因为没有包含噪声时)频域中的总功率是时域中总功率的一半,因为我已经切断了负频率。

然而,随着时域信号的噪声增加,信号+噪声的傅立叶变换的总功率远小于信号+噪声总功率的一半。

我的代码如下:

import numpy as np
import numpy.fft as nf
import matplotlib.pyplot as plt

def findingdifference(randomvalues):  

    n               = int(1e7) #number of points
    tmax            = 40e-3 #measurement time
    f1              = 30e6 #beat frequency

    t               = np.linspace(-tmax,tmax,num=n) #define time axis
    dt              = t[1]-t[0] #time spacing

    gt              = np.sin(2*np.pi*f1*t)+randomvalues #make a sin + noise

    fftfreq         = nf.fftfreq(n,dt) #defining frequency (x) axis
    hkk             = nf.fft(gt) # fourier transform of sinusoid + noise
    hkn             = nf.fft(randomvalues) #fourier transform of just noise

    fftfreq         = fftfreq[fftfreq>0] #only taking positive frequencies
    hkk             = hkk[fftfreq>0]
    hkn             = hkn[fftfreq>0]

    timedomain_p    = sum(abs(gt)**2.0)*dt #parseval's theorem for time
    freqdomain_p    = sum(abs(hkk)**2.0)*dt/n # parseval's therom for frequency
    difference      = (timedomain_p-freqdomain_p)/timedomain_p*100 #percentage diff

    tdomain_pn   = sum(abs(randomvalues)**2.0)*dt #parseval's for time
    fdomain_pn   = sum(abs(hkn)**2.0)*dt/n # parseval's for frequency
    difference_n    = (tdomain_pn-fdomain_pn)/tdomain_pn*100 #percent diff

    return difference,difference_n

def definingvalues(max_amp,length):

    noise_amplitude     = np.linspace(0,max_amp,length) #defining noise amplitude
    difference          = np.zeros((2,len(noise_amplitude)))
    randomvals          = np.random.random(int(1e7)) #defining noise

    for i in range(len(noise_amplitude)):
        difference[:,i] = (findingdifference(noise_amplitude[i]*randomvals))

    return noise_amplitude,difference

def figure(max_amp,length):

    noise_amplitude,difference = definingvalues(max_amp,length)

    plt.figure()
    plt.plot(noise_amplitude,difference[0,:],color='red')
    plt.plot(noise_amplitude,difference[1,:],color='blue')
    plt.xlabel('Noise_Variable')
    plt.ylabel(r'Difference in $\%$')
    plt.show()

    return
figure(max_amp=3,length=21)

我的最终图表如下figure。在我这样做的时候我做错了什么?这种趋势是否会出现增加噪音的物理原因?是否对不完美的正弦信号进行傅里叶变换?我这样做的原因是为了理解一个非常嘈杂的正弦信号,我有真正的数据。

1 个答案:

答案 0 :(得分:1)

如果你使用整个频谱(正负)频率来计算功率,Parseval定理就会成立。

出现差异的原因是DC( f = 0)组件,这有点特殊。

首先,DC组件来自哪里?您使用np.random.random生成介于0和1之间的随机值。因此,平均而言,您将信号提高0.5 * noise_amplitude,这需要大量功率。此功率在时域中正确计算。

然而,在频域中,只有一个FFT bin对应于f = 0。所有其他频率的功率分布在两个箱中,只有DC功率包含在一个箱中。

通过缩放噪音,您可以添加直流电源。通过去除负频率,可以消除一半的信号功率,但大部分噪声功率都位于完全使用的直流分量中。

您有几种选择:

  1. 使用所有频率计算功率。
  2. 使用没有DC分量的噪音:randomvals = np.random.random(int(1e7)) - 0.5
  3. 通过移除一半直流电源来“修复”功率计算:hkk[fftfreq==0] /= np.sqrt(2)
  4. 我选择选项1.第二种可能没问题,我不建议使用3。

    最后,代码存在一个小问题:

    fftfreq         = fftfreq[fftfreq>0] #only taking positive frequencies
    hkk             = hkk[fftfreq>0]
    hkn             = hkn[fftfreq>0]
    

    这没有多大意义。更好地改变它

    hkk             = hkk[fftfreq>=0]
    hkn             = hkn[fftfreq>=0]
    

    或完全删除选项1。