利用python实现matlab中的agwn(signal,SNR)函数

import numpy as np
import matplotlib.pyplot as plt

r = np.linspace(0.000001,20)
# print(r)
snr = 10*np.log10(r)
plt.plot(r,snr)
plt.grid()
plt.xlabel('$10*log_{10}\\frac{\sum{x^2}}{\sum{n^2}}$')
plt.ylabel('SNR')
plt.title('The ratio-snr curve')
plt.tight_layout()
plt.show()