使用scipy odeint遇返回数组尺寸不匹配错误，求排查与解决方法

如何查看func函数返回数组的尺寸？ #

• 在func函数里加打印语句，比如print(np.shape(dy_dt))或者print(len(dy_dt))，然后手动调用func：给它传和y0同尺寸的测试数组，比如func(y0, 0)（0是随便选的时间值），运行后就能看到返回值的尺寸。
• 用调试工具，在func的return语句前打个断点，运行时直接查看返回变量的形状。
• 手动调用func并检查：在调用odeint之前，执行result = func(y0, 0)，然后用np.shape(result)查看返回数组的尺寸。

解决尺寸不匹配问题 #

这个错误的核心是：odeint要求func返回的导数数组，必须和初始状态数组y0的尺寸完全一致——y0有多少个元素，func就得返回多少个对应的导数值。解决方向如下：

• 核对func的计算逻辑：y0是13维，就必须给每个维度的变量都计算对应的导数，不能漏算某个变量的导数，也不能多算额外的值。
• 检查返回值的构造：如果是手动写数组（比如np.array([dy1, dy2, ...])），数清楚里面的元素个数是不是13个；如果是通过矩阵运算生成返回值，确保运算结果的维度正确，别把二维数组误当成一维返回，或者反过来。
• 避免返回单个标量：如果不小心返回了一个标量（比如直接return某个单个的计算值），尺寸就会是1，自然和13不匹配。这种情况要把所有13个导数值整理成数组再返回。
• 提前测试：在调用odeint之前，先手动验证func的返回尺寸和y0一致，没问题再执行odeint调用。

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