Wählen Sie die zu lösende Variable aus (Python-Root-Suche mit fsolve)

Question

Scipy bietet viele nützliche Tools für die Root-Suche, insbesondere fsolve. Typischerweise hat ein Programm, das folgende Formular:

def eqn(x, a, b): 
    return x + 2*a - b**2 

fsolve(eqn, x0=0.5, args = (a,b)) 

und eine Wurzel für eqn(x) = 0 gegeben einige Argumente a und b finden.

Allerdings, was ist, wenn ich ein Problem habe, wo ich für die a Variable lösen will, geben die Funktion Argumente in x und ? Natürlich könnte ich die anfängliche Gleichung umformulieren, aber das scheint langatmig und ineffizient zu sein. Gibt es eine Möglichkeit, einfach fsolve (oder einen anderen Wurzelfindungsalgorithmus) zu setzen, damit ich wählen kann, für welche Variable ich auflösen möchte?

Answer 1

Sie können mit Ihrer ersten Idee in prägnanter Weise gehen lambda functions mit:

fsolve(lambda a,x,b: eqn(x, a, b), x0=0.5, args=(x,b)) 

Das heißt, die Argumente in dem Lambda-Wrapper neu anordnen, so dass Sie keine separates def eqn2(a,x,b) zu schreiben.