python - 如何使用python计算信噪比

Question

亲爱的专家，我有一个数据集。我只想计算数据的信噪比。数据在这里加载 https://i.fluffy.cc/jwg9d7nRNDFqdzvg1Qthc0J7CNtKd5CV.html
我的代码如下:

import numpy as np
from scipy import signaltonoise
import scipy.io
dat=scipy.io.loadmat('./data.mat')
arr=dat['dn']
snr=scipy.stats.signaltonoise(arr, axis=0, ddof=0) 

但我收到类似 importError 的错误:无法从 'scipy' 导入名称 'signaltonoise'
如果不存在如何计算 snr，请建议使用 python 使用此数据集使用其他方法。

Answer 1

scipy.stats.signaltonoise已在 scipy 1.0.0 中删除。您可以降级您的 scipy 版本或自己创建函数:

def signaltonoise(a, axis=0, ddof=0):
    a = np.asanyarray(a)
    m = a.mean(axis)
    sd = a.std(axis=axis, ddof=ddof)
    return np.where(sd == 0, 0, m/sd)

来源:https://github.com/scipy/scipy/blob/v0.16.0/scipy/stats/stats.py#L1963
有关文档字符串，请参阅 github 链接。
编辑:完整的脚本如下所示

import numpy as np
import scipy.io

def signaltonoise(a, axis=0, ddof=0):
    a = np.asanyarray(a)
    m = a.mean(axis)
    sd = a.std(axis=axis, ddof=ddof)
    return np.where(sd == 0, 0, m/sd)

dat = scipy.io.loadmat('./data.mat')
arr = dat['dn']
snr = signaltonoise(arr)