RK4方法Python代码报错：IndexError: index 1超出轴0大小1的范围

首先把这个错误给你讲透：这个错误直白来说就是你在访问一个只有1个元素的一维数组（或者被包装成数组的标量）时，试图取索引为1的位置，但数组索引是从0开始的，它最大的合法索引就是0，所以取1肯定越界了。

结合你写RK4的场景，最常见的问题出在这几个地方，咱一一拆解：

常见原因1：存储迭代结果的数组长度不够 #

比如你可能只初始化了一个包含初始值的数组（比如ys = np.array([y0])），但RK4是迭代计算的，每一步都要往数组里新增结果。当第一次迭代想给ys[1]赋值时，这个数组当前只有1个元素（索引0），自然就触发了这个报错。

解决办法： #

提前计算好总迭代步数，初始化一个长度足够的数组。比如：

n_steps = int((t_end - t_start) / h)
ys = np.zeros(n_steps + 1)  # 包含初始值，所以长度是步数+1
ys[0] = y0  # 把初始值放到第一个位置

这样后续迭代时，ys[i+1]就有合法的位置可以存储新结果了。

常见原因2：误将标量当成数组索引 #

如果你的初始值y0是单个数字（标量），但你在代码里某个地方写了y[1]试图索引它的第二个元素，这肯定会报错——标量根本没有索引可言。

解决办法： #

检查你定义的微分方程函数，以及RK4循环里的每一步，确认所有索引操作的对象都是足够长度的数组，而不是标量。如果是求解单变量微分方程，大部分时候变量都是标量，直接用变量本身就行，不需要索引。

给你一个能跑通的RK4示例代码 #

对比一下你自己的代码，看看哪里有差异：

import numpy as np

def rk4(f, y0, t_start, t_end, h):
    # 计算总迭代步数
    n_steps = int((t_end - t_start) / h)
    # 生成时间点数组
    t = np.linspace(t_start, t_end, n_steps + 1)
    # 初始化结果数组，长度足够容纳所有迭代值
    ys = np.zeros(n_steps + 1)
    ys[0] = y0
    
    for i in range(n_steps):
        # 计算RK4的四个k值
        k1 = h * f(t[i], ys[i])
        k2 = h * f(t[i] + h/2, ys[i] + k1/2)
        k3 = h * f(t[i] + h/2, ys[i] + k2/2)
        k4 = h * f(t[i] + h, ys[i] + k3)
        # 更新下一个时刻的y值
        ys[i+1] = ys[i] + (k1 + 2*k2 + 2*k3 + k4)/6
    
    return t, ys

# 测试用的一阶微分方程：dy/dt = -y
def test_diff_eq(t, y):
    return -y

# 调用RK4求解
t_vals, y_vals = rk4(test_diff_eq, 1.0, 0, 5, 0.1)
print(y_vals)

快速排查小技巧 #

遇到这种索引错误，最快定位问题的方式：

• 找到报错的那一行代码，把被索引的数组打印出来，看看它的形状（用print(arr.shape)），确认是不是只有1个元素。
• 回溯数组初始化的地方，检查长度是否计算正确。

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