Bỏ qua rễ tưởng tượng trong sympy

Question

Tôi đang sử dụng sympy để giải quyết một đa thức:

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

y là một danh sách các giải pháp khả thi. Tuy nhiên, tôi cần phải bỏ qua những hình ảnh ảo và chỉ sử dụng các giải pháp thực tế. Ngoài ra, tôi muốn giải pháp như một giá trị không phải là một biểu thức. Ngay bây giờ có vẻ như:

[-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] 

Tôi cần giá trị của biểu thức cuối cùng (-2.22756). Có chức năng nào trong sympy để đơn giản hóa điều này không?

Answer 1

Như Krastonov đã đề cập mpmath cung cấp một phương pháp dễ dàng hơn:

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

Tôi đã quản lý đơn giản là bỏ qua các giải pháp chứa ký tự "I" và sử dụng .evalf() để đánh giá biểu thức. Mã này bây giờ là:

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

Nếu bạn đặt x là thật, SymPy sẽ chỉ cung cấp cho bạn các giải pháp thực

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

Answer 4

solve() không có đầu ra nhất quán cho các loại giải pháp khác nhau, vui lòng sử dụng 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

Đây chính là loại điều mà real_roots được thực hiện cho và đặc biệt áp dụng đối với trường hợp của bạn, nơi các hệ số là các số nguyên:

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(...), ...] 

Giá trị của trường CRootOf có thể được đánh giá với bất kỳ độ chính xác nào bạn cần và không được chứa bất kỳ phần ảo nào. Ví dụ,

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

Lưu ý: Khi tôi gọi lại, giải quyết sẽ gửi lại rễ rằng nó đã không thể xác nhận đã gặp gỡ các giả định (tức là nếu chúng không được tìm thấy là sai đối với giả định sau đó họ được trả về). Ngoài ra, nếu bạn muốn sản lượng nhất quán hơn từ giải quyết, @PyRick, hãy đặt cờ dict=True.