Rössler Strange Attractor with Python: Part 1
Categories:
3 minute read
Introduction
One of the most popular and widely known strange attractors is the Rössler strange attractor. We will explore this mathematical object using Python in a series of blog posts. Before we go into any detail about what it is, what is a strange attractor, and so on, let’s first take a quick look at it using some simple Python code.
Simple Integration using Euler’s method
This is a quick intro, and hopefully part of a longer series on using Python to analyze the Rössler Strange Attractor.
We begin by importing the basic libraries that we will use as follows:
import numpy as np
import matplotlib
%matplotlib inline
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
Define Rössler System
Next we define a python function which will define the Rössler equations for us to integrate and graph. The Rössler equestions are
$$ \dot{x} = -y -z $$ $$ \dot{y} = x + ay $$ $$ \dot{z} = b + z(x-c) $$
They function over a number of parameters, something we will explore in more detail in future posts. For now, let’s just pick a pretty common set of parameters:
def rossler(x, y, z, a=0.2, b=0.2, c=5.7):
x_dot = - y - z
y_dot = x + a*y
z_dot = b + z*(x-c)
return x_dot, y_dot, z_dot
Numerical Integration
These equations can’t be solved for non-trivial sets of parameters, so we must integrate them numerically. Here, we use a very simpler Euler’s method of integration to get us out the door. This method is never a good idea to use for non-trivial applications, but is a quick easy way to get started.
Euler’s method makes use of the definition of a derivative:
$$\dot{x} = \frac{dx}{dt} = \lim_{\Delta \rightarrow 0} \frac{\Delta x}{\Delta t} $$
To create a simplistic way to numerical integrate where:
$$ \Delta x = \dot{x} \Delta t $$
This isn’t very accurate over the long term, but with small enough $\Delta t$ can give a fair approximation in the short term. For this simple demo, we choose to let $dt = 0.05 s$ and choose to plot 1000 points for a total of 50 seconds of data.
dt = 0.05
stepCnt = 1000
Next we set up NUMPY arrays that will hold the 1000 points for each dimension, and then set intitial values (what initial values to use will be explained in more detail in a post where we discuss the parameters as well).
# Need one more for the initial values
xs = np.empty((stepCnt + 1,))
ys = np.empty((stepCnt + 1,))
zs = np.empty((stepCnt + 1,))
# Setting initial values
xs[0], ys[0], zs[0] = (0.1, 1., 1.05)
Next, we run the Euler’s method integration to populate the arrays:
# Stepping through "time".
for i in range(stepCnt):
# Derivatives of the X, Y, Z state
x_dot, y_dot, z_dot = rossler(xs[i], ys[i], zs[i])
xs[i + 1] = xs[i] + (x_dot * dt)
ys[i + 1] = ys[i] + (y_dot * dt)
zs[i + 1] = zs[i] + (z_dot * dt)
Plot
Our final step is to plot the results so that we can see the trajectory
of the Rössler system over these 50 seconds of integration.
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot3D(xs, ys, zs, 'gray')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Rössler')