调整ODE系统参数模拟种群T演化出现IndexError问题求助

错误原因排查 #

• 1. ODE模型函数参数使用错误：scipy.integrate.odeint传入的模型函数第一个参数为当前时刻的系统状态向量，你代码中模型函数第一个入参命名为T，但函数内部全程调用未定义的外部变量sys，完全未使用输入的状态向量。如果外部sys变量长度不足4，就会触发索引越界的IndexError。
• 2. 四元系统初始状态未正确定义：四元耦合系统对应的初始状态sys0应为长度为4的数组，你代码中仅定义了种群T的初始值T0=500，其余三个状态的初始值未补充，若sys0长度不足4，传入odeint后状态向量索引到2、3位置时也会触发越界错误。
• 3. 额外参数未定义：模型中用到的a1、a2、b2、md、l1参数未在代码中赋值，修复索引错误后运行也会触发名称报错。

修正后的代码示例 #

import numpy as np
from scipy.integrate import odeint
import matplotlib.pyplot as plt

# 先定义所有固定参数，根据你的实际需求调整数值
a1 = 0.01
a2 = 0.02
b2 = 0.001
md = 0.05
l1 = 0.01

# 四元耦合系统完整模型，输出所有四个状态的导数
def model(sys, t, d):
    s0, s1, s2, s3 = sys # 解包四个状态，s1对应种群T
    # 你给出的T的导数方程
    dTdt = s1*(a2*d - b2*s1) - a1*s2*s1 - md*s1 + l1*s0*s3
    # 补充另外三个状态的导数方程，根据你的实际系统填写，下方为占位符，替换为真实方程即可
    ds0dt = 0 
    ds2dt = 0 
    ds3dt = 0 
    return [ds0dt, dTdt, ds2dt, ds3dt]

# 定义初始状态，四个值分别对应四个状态的初始值，s1位置填你定义的T0=500
sys0 = [100, 500, 50, 20] # 其余三个初始值按你的实际需求修改
t = np.linspace(0, 100, 1000) # 增加点数让曲线更平滑

# 求解不同d下的系统演化
d = 0.1
T1 = odeint(model, sys0, t, args=(d,))
d = 0.2
T2 = odeint(model, sys0, t, args=(d,))
d = 0.6
T3 = odeint(model, sys0, t, args=(d,))

# 绘图，仅提取第二个维度（索引1）即种群T的数值
plot2 = plt.figure(2)
plt.plot(t, T1[:,1], 'r-', linewidth=2, label='d=0.1')
plt.plot(t, T2[:,1], 'b--', linewidth=2, label='d=0.2')
plt.plot(t, T3[:,1], 'g:', linewidth=2, label='d=0.6')
plt.xlabel('时间')
plt.ylabel('种群T数量')
plt.legend()
plt.show()

额外注意点 #

odeint返回的是所有时刻所有状态的数组，形状为(时间点数量, 状态数量)，你要提取种群T的演化数据需要取第二列[:,1]，直接用整个数组绘图也会触发维度不匹配的问题。

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