实现Runge-Kutta 4方法时遇numpy数组索引错误求助

问题描述 #

实现Runge-Kutta 4（RK4）方法时，遇到索引错误：too many indices for array: array is 1-dimensional, but 2 were indexed，错误触发在while循环的u[n,1]处。初始传入的数组u是形状为(1,4)的二维数组，在函数外可正常使用二维索引，但进入函数后执行追加操作后出现问题。

错误原因 #

核心问题出在np.append的使用方式：

• 初始u是(1,4)的二维数组，但np.append会将追加的元素扁平化处理，把新的4维向量直接拼接到原数组的末尾，让u从二维变成一维数组（比如第一次追加后形状变为(8,)）。
• 后续代码尝试用u[n,1]这种二维索引访问一维数组，自然触发索引错误。

修复方案 #

使用np.vstack替代np.append，保证数组始终保持二维结构（每行是一个4维状态向量）：

import numpy as np

def RK4(u,k):
    n=0
    # 新增n < len(u)避免数组越界
    while n < len(u) and (u[n,1] >= 0.05 or n<=1):
        w1=F(n*k,u[n,:])
        w2=F(n*k+k/2,u[n,:]+k/2*w1)
        w3=F(n*k+k/2,u[n,:]+k/2*w2)
        w4=F(n*k+k,u[n,:]+k*w3) 
        # 计算新的状态向量
        new_state = u[n,:] + k/6*(w1+2*w2+2*w3+w4)
        # 用vstack追加新行，维持二维数组结构
        u = np.vstack([u, new_state])
        n+=1
    return u

# 初始化部分（假设x0,y0,vx,vy已定义）
f=np.array([x0,y0,vx,vy])
u = np.array([f])                        
print(u[-1,1])            # 输出y0
print(u.shape)            # 输出(1,4)

u_fin=RK4(u,k)            # 正常执行无错误

额外说明 #

• np.vstack会将传入的一维数组作为新行添加到二维数组中，始终保持u的形状为(N,4)，确保二维索引u[n,1]有效。
• 循环条件中新增n < len(u)，避免当n超过数组当前长度时触发索引越界错误。

内容的提问来源于stack exchange，提问作者Niken24