В sympy у меня есть интеграл, который возвращает кусочный объект, например.
In [2]: from sympy.abc import x,y,z
In [3]: test = exp(-x**2/z**2)
In [4]: itest = integrate(test,(x,0,oo))
In [5]: itest
Out[5]:
⎧ ___
⎪ ╲╱ π ⋅z │ ⎛ 1 ⎞│ π
⎪ ─────── for │periodic_argument⎜──────────────, ∞⎟│ ≤ ─
⎪ 2 │ ⎜ 2 ⎟│ 2
⎪ │ ⎝polar_lift (z) ⎠│
⎪
⎪∞
⎪⌠
⎨⎮ 2
⎪⎮ -x
⎪⎮ ───
⎪⎮ 2
⎪⎮ z
⎪⎮ ℯ dx otherwise
⎪⌡
⎪0
⎩
Я хотел бы извлечь только первую ветвь этого кусочного уравнения, другими словами, я хотел бы иметь возможность делать что-то вроде itest.parts(0)
, чтобы извлечь просто sqrt(pi)*z/2
. Кажется, я не могу найти способ сделать это, но, возможно, я использую неправильные условия поиска в документации. Любые идеи?
Изменить
Немного покопавшись, я обнаружил, что если я сделаю itest.args[0][0]
, я могу извлечь это выражение. Однако это похоже на хитрость. Есть ли лучший подход?