Time-dependent ODE

from scipy.integrate import ode

def f(t, y, arg1):
    return [1j*arg1*y[0] + y[1], -arg1*y[1]**2]
def jac(t, y, arg1):
    return [[1j*arg1, 1], [0, -arg1*2*y[1]]]

r = ode(f, jac).set_integrator('zvode', method='bdf')

t = np.arange(0, 10, 1)
results = np.empty((len(t), 1 + len(y0)), dtype=float) + np.nan

r.set_initial_value([1.0j, 2.0], t[0])
for ind in range(len(t) - 1):
    r.set_f_params(param[ind]).set_jac_params(param[ind])
    tmp_result = r.integrate(t[ind+1])
    if not r.successful():
        break
    results[ind] = tmp_result