微分方程组的动态生成

Question

我有一个odeintPython模块的ODE系统：

def equations(p, t, lmbds):
    return np.array([-p[0] * lmbds[0] + p[1] * lmbds[1],
                     -p[1] * lmbds[1] - p[1] * lmbds[2] + p[0] * lmbds[0]])

该系统对应于两态三强度的马尔可夫链

方程组可以包含不同数量的方程。是否可以根据用户输入的公式数量在移动中生成它们

Answer 1

请参见下面创建n个方程式的方法：

def get_equation(p, t, lmbds, n):
    """ Returns equation for provided n. """
    # Create and return equation

def equations(p, t, lmbds, eq_num):
    equations = []
    for n in range(eq_num):
        equation = get_equation(p, t, lmbds, n)
        equations.append(equation)
    return np.array(*equations)