Ошибка в scipy Curve_fit при использовании Absolute_sigma ⇐ Python

[Anonymous]

У меня есть этот код:
import numpy as np
from scipy.optimize import curve_fit

def lin_func(x, a, b):
return a * x + b

a = 0.005/4
b = 10000/4
N = 10000

position_list = np.array([-100,-50,-20,0,20,50,100])

def get_params(N):
counts_list = np.zeros(len(position_list))
for i in range(len(position_list)):
position_0 = position_list
counts = 0
for j in range(N):
position = np.random.normal(position_0,10)
p = 0.5+a+b*position*10**(-6)
counts = counts + np.sum(np.random.binomial(1000, p, 1))
counts_list = counts
print(np.sqrt(counts_list))
popt, pcov = curve_fit(lin_func, position_list*10**(-6), counts_list, absolute_sigma = np.sqrt(counts_list))
return popt[1]/(N*1000)

K = 100
params_list = np.zeros(K)
for i in range(K):
params_list = get_params(N)-0.5
print(params_list)

Когда я запускаю его, какое-то время он работает нормально, но затем выдает следующую ошибку:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Это происходит только в том случае, если я добавляю Absolute_sigma = np.sqrt(counts_list) к вызову функции, однако если я печатаю counts_list, я не вижу ничего плохого с номером, как раз перед появлением ошибки (они кажутся похожими на значения, напечатанные, когда вызов функции работает нормально). Я не знаю, как это отладить.
Это полное сообщение об ошибке:
ValueError Traceback (most recent call last)
Cell In[10], line 4
2 params_list = np.zeros(K)
3 for i in range(K):
----> 4 params_list = get_params(N)-0.5
5 print(params_list)

Cell In[9], line 13, in get_params(N)
11 counts_list = counts
12 print(np.sqrt(counts_list))
---> 13 popt, pcov = curve_fit(lin_func, position_list*10**(-6), counts_list, absolute_sigma = np.sqrt(counts_list))
14 return popt[1]/(N*1000)

File ~/miniforge3/lib/python3.10/site-packages/scipy/optimize/_minpack_py.py:897, in curve_fit(f, xdata, ydata, p0, sigma, absolute_sigma, check_finite, bounds, method, jac, full_output, **kwargs)
895 pcov.fill(inf)
896 warn_cov = True
--> 897 elif not absolute_sigma:
898 if ysize > p0.size:
899 s_sq = cost / (ysize - p0.size)

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()


Подробнее здесь: https://stackoverflow.com/questions/791 ... lute-sigma