Sympy arbeitet mit int params, aber spinnt mit floats

Question

Ich benutze sympy 1.0 mit Python 2.7.10 auf einem Mac. Das Problem ist, dass wenn ich Floats für Parameter anstelle von Ints verwende, scheint solve() für immer zu drehen.

Hier ist mein Skript, mit a, b und c wie int angegeben. Es gibt die folgende Ausgabe mit oder ohne a, b smpifying und c:

from sympy import * 

x = symbols('x') 
a = 500 
b = 10 
c = 333 

y = a + b * (x - c) - (a * (x/c) + x * b * log(x/c)) 

print 'y=', y 

solution = solve(y, x) 
print 'solve() gives:', solution 

Diese Drucke:

y= -10*x*log(x/333) + 2830*x/333 - 2830 
solve() gives: [333*exp(LambertW(-283*exp(-283/333)/333) + 283/333)] 

Das ist richtig, wie ich durch die Lösung numerisch mit nSolve prüft haben() und Vergleichen mit einer unabhängig berechneten Lösung.

Nun ändern Sie die Definition von a, b und c:

a = 500.0 
b = 10.0 
c = 333.0 

Dann wird die Ausgabe wie folgt:

y= -10.0*x*log(0.003003003003003*x) + 8.4984984984985*x - 2830.0 

Damit lösen() dreht, scheinbar für immer. Beachten Sie, dass dieser Ausdruck numerisch korrekt angezeigt wird.

Also noch einmal, die Frage ist: Wie kann ich Gleitkommaparameter in sympy Gleichungen verwenden?

Answer 1

Ich kann nicht sagen, warum Sympy Probleme hat, aber ich vermute, dass es damit zu tun hat, dass die Gleitkommadarstellung nicht exakt ist.

Eine Abhilfe zu solve sein würde, ohne explizit die Parameter spezifiziert a, b und c, und verwenden Sie dann .subs()

import sympy as sp 
x, a, b, c = sp.symbols('x, a, b, c') 
y = a + b * (x - c) - (a * (x/c) + x * b * sp.log(x/c)) 
sol = sp.solve(y, x) 

# find the second root manually by considering the k=-1 branch 
# (solve provides only the k=0 branch) 
sol_two_branches = [sol[0], sol[0].replace(sp.LambertW, lambda *args: sp.LambertW(*args,-1))] 

print([s.subs({a:500,b:10,c:333}).n() for s in sol_two_branches]) 

[333.000000000000, 242.528588686908] 

PS: Kredit für infroming mich über die Zweigauswahl an @asmeuer geht Option von LambertW

Answer 2

Ich glaube, das Problem ist, dass lösen Konvertierungen floats zu rationals standardmäßig, die lea ds zu enormen Kräften, wenn das Protokoll invertiert wird. solve(rational=False) sollte dies ausschalten, aber für mich verursacht dies solve mit NotImplementedError zu scheitern. Ich öffnete an issue dafür.