リストの要素（数値）を取得

Question

現在、sympyを使って次の方程式を解いてます。codeは次の通りです。

x = sp.Symbol('x')
x0 = sp.solve(x**3 - 2*x -1, x)

これを解くと、x0はリストとして返ってくるのですが、これを数値として取り出すにはどうすればよいでしょうか。
できれば、float型で取り出したいです。

Answer 1

こういうことでしょうか？

import sympy as sp

x = sp.Symbol('x')
x0 = sp.solve(x**3 - 2*x -1, x)
print(x0)
# [-1, 1/2 - sqrt(5)/2, 1/2 + sqrt(5)/2]

x0 = [float(f) for f in x0]
print(x0)
# [-1.0, -0.6180339887498949, 1.618033988749895]

※コメントアウトされているものはprint関数の出力を意味します。

分かりずらいところありましたら気軽に聞いてください。

Comments

[hamberger]

まさに欲しかった答えでした！リスト内包表記というものを使われているのですね。
夜遅くにありがとうございました！

[rtgsdfsdg]

リスト内包表記を使うと処理速度向上が期待できるのでお勧めです