Согласно scipy.optimize.newton ваш x0
должен быть скаляром, а не массивом или кортежем (то, что вы передаете scipy.optimize.newton()
в своем коде). К счастью, scipy.optimize.fsolve также может найти нулевую позицию и принимает массивы как x0
.
Из того, что я вижу, вы усложняете жизнь, используя sympy
, когда вы можете просто использовать numpy
, что ускоряет ваш код (вы можете избежать цикла for
). В следующем примере показано, как его можно использовать (надеюсь, это поможет):
import warnings
import numpy as np
import scipy.optimize as opt
# filter warnings
warnings.filterwarnings("ignore")
data = np.array([3, 33, 146, 227, 342, 351, 353, 444, 556, 571, 709, 759, 836, 860, 968,
1056, 1726, 1846, 1872, 1986, 2311, 2366, 2608, 2676, 3098, 3278, 3288,
4434, 5034, 5049, 5085, 5089, 5089, 5097, 5324, 5389,5565, 5623, 6080,
6380, 6477, 6740, 7192, 7447, 7644, 7837, 7843, 7922, 8738, 10089,
10237, 10258, 10491, 10625, 10982, 11175, 11411, 11442, 11811, 12559,
12559, 12791, 13121, 13486, 14708, 15251, 15261, 15277, 15806, 16185,
16229, 16358, 17168, 17458, 17758, 18287, 18568, 18728, 19556, 20567,
21012, 21308, 23063, 24127, 25910, 26770, 27753, 28460, 28493, 29361,
30085, 32408, 35338, 36799, 37642, 37654, 37915, 39715, 40580, 42015,
42045, 42188, 42296, 42296, 45406, 46653, 47596, 48296, 49171, 49416,
50145, 2042, 52489, 52875, 53321, 53443, 54433, 55381, 56463, 56485,
56560, 57042, 62551, 62651, 62661, 63732, 64103, 64893, 71043, 74364,
75409, 76057, 81542, 82702, 84566, 88682])
n = len(data)
tn = data[n-1]
f = lambda b, c: n - np.sum(np.log(b * c * data**(c-1) * np.exp(-b * data**c)))
obj_func = lambda x : np.array([f(x[0], x[1]) - x[0], f(x[0], x[1]) - x[1]])
x0 = np.array([0.00404, 1.0])
sol = opt.fsolve(obj_func, x0 = x0)
print('b, c = ', sol)
Выход:
b, c = [0.01737803 0.4300348 ]