A simple runge-kutta 4 integrator
Project description
A simple implementation of Runge-Kutta for python.
Setup
Install
pip install pyrk
Develop
git clone https://github.com/walchko/pyrk cd pyrk pip install -e .
Usage
See the examples in the docs folder or a simple one:
from __future__ import division, print_function
from pyrk import RK4
import numpy as np
import matplotlib.pyplot as plt
def vanderpol(t, xi, u):
dx, x = xi
mu = 4.0 # damping
ddx = mu*(1-x**2)*dx-x
dx = dx
return np.array([ddx, dx])
rk = RK4(vanderpol)
t, y = rk.solve(np.array([0, 1]), .01, 200)
y1 = []
y2 = []
for v in y:
y1.append(v[0])
y2.append(v[1])
plt.plot(y1, y2)
plt.ylabel('velocity')
plt.xlabel('position')
plt.grid(True)
plt.show()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pyrk-0.5.4.tar.gz
(2.2 kB
view hashes)