我一直在尝试使用Runge-Kutta45积分方法来更新空间中粒子的一组位置和速度,以获得某个时间步长的新状态。
最初,我创建了一个包含所有这些元素的数组,并将它们组合在一起(y):
r_vec_1 = np.array([0, 0])
v_vec_1 = np.array([-np.sqrt(2), -np.sqrt(2)])
r_vec_2 = np.array([-1, 0])
v_vec_2 = np.array([np.sqrt(2) / 2, np.sqrt(2) / 2])
r_vec_3 = np.array([1, 0])
v_vec_3 = np.array([np.sqrt(2) / 2, np.sqrt(2) / 2])
y_0 = np.concatenate((r_vec_1, v_vec_1, r_vec_2, v_vec_2, r_vec_3, v_vec_3))
y = y_0
现在,我用这个数组作为我的初始条件并创建了一个函数它给了我一个新的函数F(y)它是函数y的导数用一阶ode集合表示
def fun(t,y):
np.array([y[2], y[3], x1_double_dot(y, mass_vector), y1_double_dot(y, mass_vector),
y[6], y[7], x2_double_dot(y, mass_vector), y2_double_dot(y, mass_vector),
y[10], y[11], x3_double_dot(y, mass_vector), y3_double_dot(y, mass_vector)])
一旦我获得了新的函数文件,我使用了初始和最终时间以及在scipy. integration . rk45子例程中需要的时间步长,产生了以下代码:
#Time start, step, and finish point
t_0 = 0
t = 0
t_step = 0.01
t_final = 200
nsteps = int((t_final - t)/t_step)
#The loop for running the Runge-Kutta method over some time period.
for step in np.linspace(t, t_final, num = nsteps):
y_new = sp.integrate.RK45(fun(t,y), t_0, y_0, t_final,vectorized=True)
history.append(y_new)
y_new = y
t += dt
history = np.array(history)
问题是,一旦我运行代码,我希望函数y更新到新的状态,并在一段时间内保持集成,直到耗尽。但是,在运行此命令后,我收到以下错误消息:
Traceback (most recent call last):
File "C:UsersRSlatPycharmProjectspythonProjectPractice3BP Calculator.py", line 68, in <module>
y_new = sp.integrate.RK45(fun(t,y), t_0, y_0, t_final,vectorized=True)
File "C:UsersRSlatPycharmProjectspythonProjectPracticelibsite-packagesscipyintegrate_ivprk.py", line 94, in __init__
self.f = self.fun(self.t, self.y)
File "C:UsersRSlatPycharmProjectspythonProjectPracticelibsite-packagesscipyintegrate_ivpbase.py", line 138, in fun
return self.fun_single(t, y)
File "C:UsersRSlatPycharmProjectspythonProjectPracticelibsite-packagesscipyintegrate_ivpbase.py", line 125, in fun_single
return self._fun(t, y[:, None]).ravel()
File "C:UsersRSlatPycharmProjectspythonProjectPracticelibsite-packagesscipyintegrate_ivpbase.py", line 20, in fun_wrapped
return np.asarray(fun(t, y), dtype=dtype)
TypeError: 'NoneType' object is not callable
任何帮助都将是非常感激的。谢谢,祝你有美好的一天!
显然(并且根据https://docs.scipy.org/doc/scipy/reference/generated/scipy.integrate.RK45.html) sp. integration . rk45()在第一个位置需要一个可调用对象。
因此,当你这样写时,它应该可以工作:
sp.integrate.RK45(fun, t_0, y_0, t_final,vectorized=True)
如你所见,我只给函数(callable) "fun"(无参数)到RK45