噪声数据中的渐变,python

2022-04-12 00:00:00 python smoothing gradient

问题描述

我有一个宇宙射线探测器的能谱。频谱遵循指数曲线,但其中会有广泛的(也许是非常轻微的)块。显然,这些数据包含了一些噪声元素。

我正在尝试平滑数据,然后绘制其渐变曲线。 到目前为止,我一直使用scipy sline函数对其进行平滑处理,然后使用np.gradient()。

从图中可以看到,梯度函数的方法是找出每个点之间的差异,它没有很清楚地显示肿块。

我基本上需要一个平滑的梯度图。任何帮助都是令人惊叹的!

我尝试了两种样条法:

def smooth_data(y,x,factor):
    print "smoothing data by interpolation..."
    xnew=np.linspace(min(x),max(x),factor*len(x))
    smoothy=spline(x,y,xnew)
    return smoothy,xnew

def smooth2_data(y,x,factor):
    xnew=np.linspace(min(x),max(x),factor*len(x))
    f=interpolate.UnivariateSpline(x,y)
    g=interpolate.interp1d(x,y)
    return g(xnew),xnew

编辑:已尝试数字微分:

def smooth_data(y,x,factor):
    print "smoothing data by interpolation..."
    xnew=np.linspace(min(x),max(x),factor*len(x))
    smoothy=spline(x,y,xnew)
    return smoothy,xnew

def minim(u,f,k):
    """"functional to be minimised to find optimum u. f is original, u is approx"""
    integral1=abs(np.gradient(u))
    part1=simps(integral1)
    part2=simps(u)
    integral2=abs(part2-f)**2.
    part3=simps(integral2)
    F=k*part1+part3
    return F


def fit(data_x,data_y,denoising,smooth_fac):
    smy,xnew=smooth_data(data_y,data_x,smooth_fac)
    y0,xnnew=smooth_data(smy,xnew,1./smooth_fac)
    y0=list(y0)
    data_y=list(data_y)
    data_fit=fmin(minim, y0, args=(data_y,denoising), maxiter=1000, maxfun=1000)
    return data_fit

但是,它只是再次返回相同的图形!


解决方案

上面发布了一个有趣的方法:Numerical Differentiation of Noisy Data。它应该会为你的问题提供一个很好的解决方案。更多细节见另一篇accompanying paper。作者还提供了Matlab code that implements it;也可以选择implementation in Python。

如果您想使用样条法,我建议调整scipy.interpolate.UnivariateSpline()的平滑系数s

另一种解决方案是通过卷积使函数平滑(例如使用高斯函数)。

我链接到的论文声称要防止卷积方法产生的一些伪影(样条法可能会遇到类似的困难)。

相关文章