信噪比python代码实现

1 定义

原始信号：x
噪声信号：n
信噪比：SNR

单位是db

 其中Psignal是原信号能量（或者是功率），Pnosie是噪声能量（或者是功率）

 

import numpy as np
import matplotlib.pyplot as plt

r = np.linspace(0.000001,20)
snr = 10*np.log10(r)

plt.plot(r,snr)
plt.grid()
#plt.xlabel('$\farc{{\sum{x^2}}{\sum{n^2}}}$')
plt.xlabel('$10*log_{10}\\frac{\sum{x^2}}{\sum{n^2}}$')
plt.ylabel('SNR')
plt.title('The ratio-snr curve')
plt.tight_layout()

 

'''
程序中用hist()检查噪声是否是高斯分布，psd()检查功率谱密度是否为常数。
'''
import numpy as np
import pylab as plt

def wgn(x, snr):
    snr = 10**(snr/10.0)
    xpower = np.sum(x**2)/len(x)
    npower = xpower / snr
    return np.random.randn(len(x)) * np.sqrt(npower)

plt.figure()
t = np.arange(0, 10000000) * 0.1
x = np.sin(t)
n = wgn(x, 6)
xn = x+n # 增加了6dBz信噪比噪声的信号
plt.subplot(221)
plt.plot(t[:1024],x[:1024])
plt.title('The original signal-x')

plt.subplot(222)
plt.plot(t[:1024],xn[:1024])
plt.title('The original sinal with Gauss White Noise')

plt.subplot(223)
plt.hist(n, bins=100, normed=True)
plt.title('Gauss Noise Distribution')

plt.subplot(224)
plt.psd(n)
plt.title('PSD')
plt.tight_layout()

plt.savefig('show the result.png',dpi=600)