У меня есть энергетический спектр с детектора космических лучей. Спектр следует экспоненциальной кривой, но в нем будут большие (и, возможно, очень небольшие) глыбы. Данные, очевидно, содержат элемент шума.
Я пытаюсь сгладить данные, а затем построить их градиент. До сих пор я использовал функцию scipy sline, чтобы сгладить ее, а затем np.gradient().
Как вы можете видеть на рисунке, метод функции градиента заключается в том, чтобы найти различия между каждой точкой, и он не очень четко показывает глыбы.
Мне в основном нужен гладкий градиентный график. Любая помощь будет потрясающей!
Я пробовал 2 метода сплайна:
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
Однако он просто снова возвращает тот же график!