用Odeint解復矩陣微分方程

Question

我想求解一個矩陣微分方程， 像這樣一個 ：

import numpy as np
from scipy.integrate import odeint


def deriv(A, t, Ab):
    return np.dot(Ab, A)


Ab = np.array([[-0.25,    0,    0],
               [ 0.25, -0.2,    0],
               [    0,  0.2, -0.1]])

time = np.linspace(0, 25, 101)
A0 = np.array([10, 20, 30])

MA = odeint(deriv, A0, time, args=(Ab,))

但是，這在具有復雜矩陣元素的情況下不起作用。 我正在尋找類似於scipy.integrate.complex_ode東西，但需要odeint 。 如果這不可能，那么我應該使用其他哪個庫來執行集成？ 我感謝您的幫助！

Answer 1

odeintw用於包裝odeint必須以同樣的方式被用作問題。 但是，初始值A0必須是復數值向量。