Ignoruj ​​wyimaginowane korzenie w sympy

Question

Używam sympy rozwiązać wielomian:

x = Symbol('x') 
y = solve(int(row["scaleA"])*x**3 + int(row["scaleB"])*x**2 + int(row["scaleC"])*x + int(row["scaleD"]), x) 

y jest lista możliwych rozwiązań. Muszę jednak zignorować te wyobrażone i wykorzystywać tylko prawdziwe rozwiązania. Chciałbym też, aby rozwiązanie było wartością, a nie wyrażeniem. Teraz wygląda na to:

[-2/3 - 55**(1/3)*(-1/2 - sqrt(3)*I/2)/3, -2/3 - 55**(1/3)*(-1/2 + sqrt(3)*I/2)/3, -55**(1/3)/3 - 2/3] 

Potrzebuję wartości ostatniego wyrażenia (-2.227756). Czy istnieją funkcje w sympy, aby to uprościć?

Answer 1

Jak Krastonov wspomniał mpmath warunkiem łatwiejszą metodę:

y = polyroots([int(row["scaleA"]), int(row["scaleB"]), int(row["scaleC"]), int(row["scaleD"])-value]) 
for root in y: 
    if "j" not in str(root): 
     value = root 

Answer 2

Udało mi się po prostu zignorować rozwiązania zawierające znak "I" i użyć .evalf() do oceny wyrażenia. Kod jest teraz:

x = Symbol('x') 
    y = solve(int(row["scaleA"])*x**3 + int(row["scaleB"])*x**2 + int(row["scaleC"])*x + int(row["scaleD"]), x) 
    for root in y: 
     if "I" not in str(root): 
      print("This One:" + str(root.evalf())) 

Answer 3

Jeśli ustawisz x być prawdziwe, SymPy tylko daje realne rozwiązania

x = Symbol('x', real=True) 
solve(..., x) 

Answer 4

solve() nie ma spójnego wyjścia dla różnych typów rozwiązań, należy użyć solveset(Eq,x,domain=S.Reals):

from sympy import ImageSet, S 
x = Symbol('x') 
y = solveset(int(row["scaleA"])*x**3 + int(row["scaleB"])*x**2+int(row["scaleC"])*x + int(row["scaleD"]), x, domain=S.Reals) 

http://docs.sympy.org/latest/modules/solvers/solveset.html

Answer 5

To jest dokładnie ten rodzaj rzeczy, które real_roots jest dla i jest szczególnie zastosowanie do przypadku, w którym współczynniki są liczbami całkowitymi:

x = Symbol('x') 
eq = int(row["scaleA"])*x**3 + int(row["scaleB"])*x**2 + int(row["scaleC"])*x + int(row["scaleD"]) 
y = real_roots(eq, x) # gives [CRootOf(...), ...] 

Wartość przypadkach CRootOf mogą być oceniany według potrzebnej precyzji i nie powinien zawierać żadnej części urojonej. Na przykład,

>>> [i.n(12) for i in real_roots(3*x**3 - 2*x**2 + 7*x - 9, x)] 
[1.07951904858] 

Uwaga: O ile pamiętam, rozwiązywać odeśle korzenie, że nie był w stanie potwierdzić spełnione założenia (czyli jeśli nie zostały one uznane za fałszywe na założeniu, następnie są one zwracane). Ponadto, jeśli chcesz bardziej spójne wyjście z rozwiązania, @PyRick, ustaw flagę dict=True.