Python-振动信号加入噪声-代码实现

好人就是拉风

于 2021-08-08 11:40:36 发布

阅读量2.2k

点赞数 1

文章标签： python noise 信号处理

本文链接：https://blog.youkuaiyun.com/qq_41645987/article/details/119512337

版权

本文介绍了在信号处理中，信号与噪声强度比(SNR)的概念，以及如何使用Python代码实现加随机噪声到输入振动信号的方法。通过`wgn`函数，可以模拟不同SNR条件下的噪声合成，适用于噪声抑制或信号增强的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 原理

在这里插入图片描述

2.代码

def wgn(x, snr):
    ## x: input vibration signal shape (a,b); a:samples number; b samples length
    ## snr: noise intensity,like -8,4,2,0,2,4,8
    ### snr=0 means noise equal to vibration signal 
    ### snr>0 means vibration signal stronger than noise, →∞ means no noise
    ### snr<0 means noise stronger than vibration signal  →-∞ means no signal
    Ps = np.sum(abs(x)**2,axis=1)/len(x)
    Pn = Ps/(10**((snr/10)))
    row,columns=x.shape
    Pn = np.repeat(Pn.reshape(-1,1),columns, axis=1)

    noise = np.random.randn(row,columns) * np.sqrt(Pn)
    signal_add_noise = x + noise
    return signal_add_noise