向量微分方程

Question

我试着用Python解决一个向量微分方程，但我总是得到一个错误，我不是很理解。下面是我的代码：

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

N=100

tau_s=1
tau_n=1
R=1

t0=0
t1=10

v_rest=-65

M=[]


for i in range(N):
    M.append([])
    for j in range(N):
        M[i].append(1)

M=np.array(M)

def I(t):
    I=5   
    return I

def system(u,t):
    du=np.zeros((N,)

    du=-1/tau_s*(u-v_rest)+R*(I(t)+np.dot(M,u))

    return du

u0=v_rest*np.ones(N,)

ts=(0,1000)

sol=solve_ivp(system,ts,u0)

我得到的错误是：

ValueError: could not broadcast input array from shape (100,100) into shape (100)

如果我理解正确的话，这意味着diff方程的一边与另一边的形状不同，因此Python无法求解它，但M和u的乘法应该会产生一个形状为(100)的向量，所以我不确定发生了什么。

你能帮我吗？

Answer 1

在documentation for solve_ivp中：“调用签名是有趣的(t，y)。这里t是一个标量”。我认为你在系统中的论点被调换了。

如果我把系统的签名改为system(t, u)，你的代码就能很好地运行。虽然，我不能说我是否得到了预期的答案。