如何使用FiniteSet的元素进行以下计算？

Question

我正在尝试使用solveset(方程，变量)来求解一个方程，这将得到变量的值作为结果。结果，我得到了一个值{ FiniteSet }。这个浮点值是否可访问，以便我可以在下一次计算中使用该值(希望是FiniteSet类型)？

问题如下:我定义了三个SymPy符号(drift1, drift2, drift3 = symbols('drift1 drift2 drift3'))，它们应该是用于以下计算的某种占位符。他们没有价值观。它们是两个方程(equation1和equation2，都是drift1, drift2和drift3的函数)的一部分。我的目标是编写一个迭代循环，通过求解这两个方程来优化所有三个符号drift1、drift2和drift3，并将结果用于下一个迭代步骤。为此，我定义了三个Python变量x, y和z，a和y的初始值分别为0.0188和0.0099。Z是x和y的函数，每个迭代步骤都会计算出新的z。在第一步中，用x和y(以及我在代码中定义的其他变量)计算z，并用equation1中的y和z替换drift2和drift3 (方程现在是drift1的函数)。通过对drift1激活equation1=0，我使用了solveset(equation1.subs(drift2, y).subs(drift3, z), drift1)，并获得了一个包含结果的FiniteSet。我将使用这个结果通过执行solveset(equation2.subs(drift1, result from FiniteSet).subs(drift3, z), drift2)来求解第二个方程。这不起作用，因为结果不是float变量，而是sympy.sets.sets.FiniteSet-type。

谁能告诉我如何使用值的值('inside‘the FiniteSet；{ FiniteSet })进行下一次计算？我使用Pyhton 3和SymPy。

import numpy as np
from sympy import *
import scipy as sp

drift1, drift2, drift3 = symbols('drift1 drift2 drift3')
init_printin(use_unicode=True)

equation1 =  0.005*(drift1*(-23.88*drift2*(15.16*drift3 + 1.15)-20.31*drift3
                       +00.47*drift2*(15.16*drift3+1.15)+1.03*drift3+0.05))/drift1

equation2 = 0.005*(drift1*(35.73*drift2*(-13.79*drift3 + 0.86)+7.91*drift3+2.09)
                       +1.65*drift2*(-13.79*drift3+0.86)+0.75*drift3+0.07)/drift1

x, y = 0.0188, 0.0099
gesamtlaenge = 1.676
laenge1 = 0.04
laenge2 = 0.02

# the loop for the iteration is not included in this code
# the only thin I wanna know is how to substitute drift1 by the solution of 
# the first calculation sol1

z = gesamtlaenge-laenge1-laenge2-x-y
expr1 = equation1.subs(drift2, y).subs(drift3, z)
sol1 = solveset(equation, drift1)
# up to this point, the code is running, but now, I get a FiniteSet {0.047} as
# result for sol1 and I wanna use this value to substitute drift1 in the next calculation
expr2 = equation2.subs(drift1, sol1).subs(drift3, z)
sol2 = solveset(equation, drift2)
print(drift1, drift2, drift3)

当我执行expr2 = equation2.subs(drift1, sol1).subs(drift3, z)时，我得到equation2作为drift1和drift2的函数，因为它不能使用FiniteSet {0,047}中的值(0.047是我需要的值)。我理解为什么会这样，但我没有解决办法。

我非常感谢您的帮助！干杯

Answer 1

您可以通过转换为列表来访问FiniteSet的元素。在这种情况下，只要有一个解决方案，list(sol1).pop()就会起作用：

>>> sol1 = FiniteSet(1)
>>> x.subs(x, list(sol1).pop())
1