Python中如何更新参数和传递来解决_ivp

Question

在我的ODE函数中，我需要迭代求解一个参数方程，直到在每个时间步收敛。我想传递最新的参数值作为下一个时间步的初始值，这样当函数迭代更新参数时，所需的时间会更少。但我不知道该怎么做。ODE函数的代码结构如下所示：

    from scipy.integrate import solve_ivp

    def run(t, y):
      if t==0:
        a = 1e-8
      nn = 0
      while nn<=100:
        nn = nn +1
        #update a until convergence
    return a*y

在某种语言中，我可以返回更新后的参数供积分器使用，但我不知道solve_ivp如何实现这一点

Answer 1

我想你要找的是以下形式的东西：

class test:
    a = 1e-8
    def f(self, t, y):
        ## do iter on self.a
        return self.a*y
        
t = test()
# solve_ivp(t.f, .....)

这样，您就可以始终使用a的最后一个值，因为它是test类实例的一部分。这并不完全是您所要求的，因为每次solve_ivp计算f时都会调用迭代，这将是每个时间步的多次。但是，我认为这是最接近的，因为solve_ivp似乎没有callback函数在每个时间步之后调用