diffeqzoo.vector_fields Module#

Vector fields for differential equations.

Functions#

affine_dependent(u, /, A, b)

Affine ODE.

affine_independent(u, /, a, b)

Affine ODE.

bratu(u, /, k)

Bratu's problem.

bratu_with_unused_derivative_argument(u, _, /, k)

Bratu's problem with signature (u, u').

fitzhugh_nagumo(u, /, a, b, c)

FitzHugh--Nagumo model.

goodwin(u, /, r, a1, a2, alpha, k)

Goodwin oscillator.

heat_1d_dirichlet(y, weights, coeff)

Discretized heat equation with Dirichlet (i.e.

henon_heiles(u, /, p)

Henon-Heiles dynamics as a second-order differential equation.

henon_heiles_first_order(u, *args)

Henon-Heiles dynamics as a first-order differential equation.

henon_heiles_with_unused_derivative_argument(u, ...)

Henon-Heiles dynamics as a second-order differential equation (with an unused second argument).

hires(u, /)

High irradiance response.

logistic(u, p0, p1, /)

Logistic ODE dynamics.

lorenz63(u, /, a, b, c)

Lorenz63 dynamics.

lorenz96(y, /, forcing)

Lorenz96 dynamics.

lotka_volterra(y, /, a, b, c, d)

Lotka--Volterra dynamics.

measles(t, u, /, mu, lmbda, eta, beta0)

Measles problem.

neural_ode_mlp(state, time, /, params)

Neural ODE based on a multi-layer perceptron.

nonlinear_chemical_reaction(u, /, k1, k2)

Nonlinear chemical reaction.

oregonator(u, /, s, q, w)

Oregonator problem.

pendulum(u, /, p)

Bratu's problem.

pendulum_with_unused_derivative_argument(u, ...)

Bratu's problem.

pleiades(u, /)

Evaluate the Pleiades vector field in its original, second-order form.

pleiades_first_order(u, *args)

The Pleiades problem as a first-order differential equation.

pleiades_with_unused_derivative_argument(u, _, /)

Evaluate the Pleiades vector field as \(\ddot u(t) = f(u(t), \dot u(t))\) (with an unused second argument).

rigid_body(y, /, p1, p2, p3)

Rigid body dynamics without external forces.

rober(u, /, k1, k2, k3)

'Rober' ODE problem.

roessler(u, /, a, b, c)

Roessler attractor.

seir(u, /, alpha, beta, gamma, population_count)

SEIR model.

sir(u, /, beta, gamma, population_count)

SIR model.

sird(u, /, beta, gamma, eta, population_count)

SIRD model.

three_body_restricted(Y, dY, /, ...)

Restricted three-body dynamics as a second-order differential equation.

three_body_restricted_first_order(u, *args)

Restricted three-body dynamics as a first-order differential equation.

van_der_pol(u, du, /, stiffness_constant)

Van-der-Pol dynamics as a second-order differential equation.

van_der_pol_first_order(u, *args)

Van-der-Pol dynamics as a first-order differential equation.