Troubels с scipy.optimize curve_fit

Question

У меня возникли проблемы с подбором кривой с помощью пакета scipy optimize. Мой код:

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

def Function_EXD_2(x, d, e):
    return d*np.exp(-x/e)

x = np.array([135, 126, 120, 100, 90, 85, 80, 70, 65, 60])
y = np.array([207, 263, 401, 460, 531, 576, 1350, 2317, 2340, 2834])

popt, pcov = curve_fit(Function_EXD_2, x, y)
print(popt, pcov)

Я получаю popt = [1,1], поэтому оптимизация не работает. Я сделал "то же самое" в R, и я собираюсь использовать popt = [44237.53, 22.21] aprox.

Может ли кто-нибудь помочь мне с этим, пожалуйста?

Большое спасибо!

Xevi

Answer 1

Есть две проблемы:

1. определение функции
2. x массив должен начинаться с 0

Я перевернул ваши данные значения и добавить границы в алгоритм подгонки

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit


def func(x, a, b, c):
    return a * np.exp(-b * x) + c


x = np.array([135, 126, 120, 100, 90, 85, 80, 70, 65, 60])
y = np.array([207, 263, 401, 460, 531, 576, 1350, 2317, 2340, 2834])

# flip array values
x = x[::-1] - np.amin(x)
y = y[::-1]

# fit function
popt, pcov = curve_fit(func, x, y, bounds=(-10**6, 10**6))

# plot data
x_data = np.linspace(1, 80, 100)
plt.plot(x, y, '.')
plt.plot(x_data, func(x_data, popt[0], popt[1], popt[2]))
plt.show()

Вывод: