抛物线镜头的 Runge-Kutta 算法中的索引越界

问题描述

您好，我正在尝试运行此程序。 z 结构被假定为大小 4，但是它给出了错误。我还注意到，如果我在 1 中打印 z 它会给我一个值，所以我不知道是什么缺陷。这是一个使用runge kutta计算物理摩擦抛物线射击系统的程序，但出现以下错误： Image of the error

代码如下：

import numpy as np
import matplotlib.pyplot as plt


def RK4(f_user,U0,t,*params):
    n=t.size
    neq=U0.size
    u=np.zeros((n,neq))
    u[0]=U0
    dt=t[2]-t[0]
    
    for i in range(n-1):
        k1=f_user(u[i],t[i],*params)
        k2=f_user(u[i]+k1*dt/2.,t[i]+dt/2.,*params)
        k3=f_user(u[i]+k2*dt/2.,*params)
        k4=f_user(u[i]+k3*dt,t[i+1],*params)
        u[i+1]=u[i]+dt*(k1+2*k2+2*k3+k4)/6.
    return u

g = 9.81
# m = 1  # kg
theta_0 = 45 * np.pi / 180  # radianes
v_0 = 10  # m/s
tInc = 0.01
tStop = (1.+ tInc)*20
t = np.arange(0.,tStop,tInc)
x_0=0
y_0=0
vx_0=v_0 * np.cos(theta_0)
vy_0=v_0 * np.sin(theta_0)
z0 = np.array([x_0,vx_0,y_0,vy_0])

def f(z,N,k,x_0,vy_0):
        v=np.sqrt(z[1] ** 2 + z[3] ** 2)
        print(z[1])
        derivs=np.array([[z[2],-k*v**(N-1)*z[2],z[3],-g-k*v**(N-1)*z[3]]])
        return derivs
psolnE = RK4(f,z0,1,.1,vy_0)```

解决方法

以下语句始终创建一个大小为 1 的 numpy 数组，其中 derivs[0] 为 [z[2],-k*v**(N-1)*z[2],z[3],-g-k*v**(N-1)*z[3]]。

derivs=np.array([[z[2],-g-k*v**(N-1)*z[3]]])

因此，一旦 RK4 中的 for 循环到达 1，就会引发错误。要解决此问题，请将上述语句替换为-

derivs=np.array([z[2],-g-k*v**(N-1)*z[3]])