Как получить несколько выходов из scipy.optimize.root? - PullRequest
Новая
       63

Как получить несколько выходов из scipy.optimize.root?

0 голосов
/ 28 июня 2019

У меня нелинейная проблема с несколькими выходами следующего типа: 

[F,out1,out2,out3] = f(x,a,b,c)

Я пытаюсь решить это в Python методом Левенберга-Марквардта способом: 

xSol = scipy.optimize.root(lambda x: f(x,a,b,c), x0, method='lm')

Если я не укажу выходных данных, т.е. F = f(x,a,b,c), метод сходится к правильному решению x, но я не могу получить другие значения моего решения, т.е. out1, out2, out3.

Согласно DOC , корневая функция не предоставляет опцию для нескольких выходов, то есть full_output=True, например, например. fsolve или bisec (что не дает хорошего решения моей проблемы).

Вот упрощенный пример задачи: 

    def root2d(x): 
        F = [np.exp(-np.exp(-(x[0] + x[1]))) - x[1] * (1 + x[0] ** 2)] 
        F.append(x[0] * np.cos(x[1]) + x[1] * np.sin(x[0]) - 0.5) 

        out = x[0]+x[1] # other results, which I would like to get too 
        return (F,out) # it yields error, but it works/converges with only "return F" 

    # --- run the test code with the root solver 
    x0 = [0,0] 
    x = root(root2d, x0) # how could I get also value of "out"? 
    print(x)
...