scipy.optimize.curve_fit производит бессмысленные подгонки кривой - PullRequest
0 голосов
/ 11 сентября 2018

Я пытаюсь подогнать кривую к некоторым сгенерированным данным, которые при построении графика напоминают экспоненциальную функцию. Я использую scipy.optimize.curve_fit, так как кажется, что он лучший (и лучше всего задокументирован) для этой работы. Фактические данные генерируются заново каждый раз, когда я запускаю код, но вот примерный набор:

import pandas
import scipy.optimize as opt

x1 = [0.4145392937447818, 0.7807888116968482, 0.7903528929788539, 
1.5081613036989836, -0.295895237606155, -0.0855307279546107, 
1.0523973736479486, -0.6967509832843239, -0.30499200990688413, 
1.1990545631966807, -1.270460772249312, 0.9531042718153095, 1.5747175535222993, 
-0.6483709650867473, 0.47820180254528477, 1.14266851615097, 0.6237953640100202, 
0.0664027559951128, 0.877280002485417, 0.9432317053343211, 1.0367424879878504, 
-0.6410400513164749, 1.667835241401498, -0.20484029870424125, 
2.887026948755316]

y1 = [0.718716626591187, 0.579938466590508, 0.722005637974309, 
1.61842778379047, 0.331301712743162, 0.342649242449043, 1.14950611092907, 
0.299221762023701, 0.345063839940754, 1.08398125906313, 0.315433168226251, 
1.3343730617376, 1.32514210008176, 0.308702648499771, 0.495749985226691, 
0.406025683910759, 0.445087968405107, 0.423578575247177, 0.816264419038205, 
1.16110461165631, 1.81572974380867, 0.420890068255763, 0.821468286117842, 
0.416275933630732, 4.7877353794036]

data = pandas.DataFrame({"Pi_values": x1, 
                         "CO2_at_solubility": y1})

Тогда я занимаюсь криволинейным бизнесом ...

##Define curve fitting
def func(x, m, c, c0):
    return c0 + m**x * c

#draw the figure
fig, ax1 = plt.subplots()
plt.xlabel('Pi Parameter')
plt.ylabel('CO2 wt%')

#plot generated data
#tried converting pandas columns to np arrays based on an issue another user was having, but it does not help
x1 = data["Pi_values"].values
y1 = data["CO2_at_solubility"].values

# Curve fitting with scipy.optimize.curve_fit
popt, pcov = opt.curve_fit(func, x1, y1)
# Use the optimized parameters to plot the best fit
plt.plot(x1, y1, 'o', x1, func(x1, *popt))

А вот очень странный результат. Независимо от того, какую форму уравнения я пробую в шутку, если она способна вписаться в любую «кривую», она выглядит так:

Click here to see the plot

Или этот беспорядок ...

Click here to see another plot

Есть идеи, что здесь может происходить? Я не смог найти других примеров, подобных этому. Я использую python3.5 в ноутбуке Jupyter.

Другие вещи, которые я пробовал, не работали: другие формы уравнения; другие уравнения; изменение начальных значений догадки; масштабирование значений в случае, если значения y были слишком малы.

Ответы [ 2 ]

0 голосов
/ 11 сентября 2018

Элементы оси x должны быть отсортированы при построении.

Пример:

x1, y1 = zip(*sorted(zip(x1, y1)))
# Curve fitting with scipy.optimize.curve_fit
popt, pcov = opt.curve_fit(func, x1, y1)
# Use the optimized parameters to plot the best fit
plt.plot(x1, y1, 'o', x1, func(x1, *popt))

, что приводит к:

enter image description here

0 голосов
/ 11 сентября 2018

Вам просто нужно отсортировать x значения с

data.sort_values(by='Pi_values', ascending=True, inplace=True)

до curve_fit:

x1 = data["Pi_values"].values
y1 = data["CO2_at_solubility"].values
# Curve fitting with scipy.optimize.curve_fit
popt, pcov = opt.curve_fit(func, x1, y1)
# Use the optimized parameters to plot the best fit
plt.plot(x1, y1, 'o', x1, func(x1, *popt))

enter image description here

...