Лоренцу подходит два способа написания кода - PullRequest
Новая
       64

Лоренцу подходит два способа написания кода

0 голосов
/ 29 мая 2018

Я сейчас борюсь с подгонкой кривой Лоренца.Я постараюсь объяснить мою проблему.Мне нужно написать собственный код для подбора кривой Лоренца, чтобы я мог добавить некоторые вещи в уравнения.Я применил Lorentzian fit с model и def, я написал аналогично, но это не работает.Проверьте мой код:

Итак, вот мои данные: 

for dataset in [Bxfft]:
    dataset = np.asarray(dataset)
    freqs, psd = signal.welch(dataset, fs=266336/300, window='hamming', nperseg=16192, scaling='density')
    plt.semilogy(freqs[30:-7000], psd[30:-7000]/dataset.size**0, color='r', label='Bx')
    x = freqs[100:-7900]
    y = psd[100:-7900]

Вот подгонка кривой Лоренца, определенная мной: 

def lorentzian(x, amp, cen, sig):
    return (amp/np.pi) * (sig/(x-cen)**2 + sig**2)

model = Model(lorentzian)
pars = model.make_params(amp=6, cen=5, sig=1)
pars['amp'].max = 6
result = model.fit(y, pars, x=x)
final_fit = result.best_fit
print(result.fit_report(min_correl=0.25))
plt.plot(x, final_fit, 'k--', linewidth=3)

И здесь сделано с помощью функции модели: 

model2 = LorentzianModel()
params2 = model2.make_params(amplitude=6, center=5, sigma=1)
params2['amplitude'].value = 6
result2 = model2.fit(y, params2, x=x)
final_fit2 = result2.best_fit
print(result2.fit_report(min_correl=0.25))
plt.plot(x, final_fit2, 'k--', linewidth=3)

Верхний график соответствует def Лоренцу, а нижний график - model Лоренцу.

И вот результат: enter image description here

1 Ответ

0 голосов
/ 29 мая 2018

Похоже на проблему с круглыми скобками.Это: 

(amp/np.pi) * (sig/(x-cen)**2 + sig**2)

не лоренц.Это: 

(amp/np.pi) * (sig/((x-cen)**2 + sig**2))

есть.Кроме того, у вас может возникнуть небольшая целочисленная проблема в редком событии cen,x,sig - все целые числа.Вы можете использовать math.pow, чтобы решить это, или что они делают в lmfit и умножить x на число с плавающей точкой: 1.0*x-cen.

В качестве примечания, lmfit по какой-то причине пишет этофункционируют аналогично, но немного по-другому (найти на странице лоренцян).Хотя я не вижу причины для этого.

...