python：满足条件时中断odeint-Java 学习之路

我正在使用scipy.integrate包中的odeint函数：

r0 = np.array([1,2,3,4])
t=np.linspace(0,1,20)
def drdt(r,t):
    return r # or whatever else
r = odeint(drdt,r0,t)

r0是一个numpy数组，包含一定数量的点的初始位置 . 在脚本结束时，正如预期的那样，我得到了20个时间点的位置 .

现在我想在满足条件r时停止odeint求解器 . 特别是当这些点中的2个比某个阈值更接近时，我想停止odeint，对r向量进行一些更改，并使用新的初始位置继续odeint求解器 . 有没有办法实现这个？

我一直在考虑的一个可能的解决方案是将odeint运行到最后，稍后检查是否满足条件，但这当然不是很有效 .

感谢大家的帮助，尼古拉

1 回答

我有一个C的答案 . 它可能不是发布它的最佳位置，但它可能仍然有趣 . （我没有找到一个更好的地方，这是我在寻找C解决方案时降落的地方） .

下面是一个C示例，当变量等于或低于零时停止集成 .

#include <iostream>
#include <boost/range/algorithm.hpp>
#include <boost/numeric/odeint.hpp>

using namespace std;
using namespace boost::numeric::odeint;


typedef double state_type;

typedef runge_kutta_cash_karp54< state_type > error_stepper_type;


class Myode{
public:
  void operator()(const state_type& x, state_type& dxdt, const double t){dxdt=-1-x;}
  static bool done(const state_type &x){return x<=0.0;}
};

int main(int argc, char* argv[]){
   Myode ode;
   state_type x=10.0;
   auto stepper =  make_controlled< error_stepper_type >( 1.0e-10 , 1.0e-6 );
   auto iter= boost::find_if(make_adaptive_range(stepper,ode,x, 0.0 , 20.0 , 0.01 ),
                 Myode::done);

   cout<<"integration stopped at"<<x<<endl;
  return 1;
}

积分在第一次达到小于或等于零的值x时停止（参见完成功能） . 因此，根据您当前的步长，它可能远远低于零 .

请注意，这使用了c 11构造，因此您需要在编译器上启用此功能 . 在我的情况下（gcc 4.4），它是通过在编译命令中添加-std = gnu 0x来实现的 .

回复于 2024-04-20T21:29:43+08:00

python：满足条件时中断odeint

1 回答

相关问题