Module Documentation

Documentation for every module and function found in TrajPy.


class trajpy.trajpy.Trajectory(trajectory=array([[0.0, 0.0]]), box_length=None, **params)[source]

Bases: object

This is the main class object in trajpy. It can be initialized as a dummy object for calling its functions or you can initialize it with a trajectory array or csv file.

__init__(trajectory=array([[0.0, 0.0]]), box_length=None, **params)[source]

Initialization function that can be left blank for using staticmethods. It can be initialized with an array with shape (N, dim) where dim is the number of spatial dimensions plus the time component. The first column must be the time, followed by the x- and y-axis. It also accepts tuples (t, x, y) or csv files.

The trajectory will be split between the temporal component self._t and the spatial axis self._r.

  • trajectory – 2D trajectory as a function of time (t, x, y)

  • params – use params for passing parameters into np.genfromtxt()

static anisotropy_(eigenvalues)[source]

Calculates the trajectory anisotropy using the eigenvalues of the gyration radius tensor.

\[a^2 = 1 - 3 \frac{\lambda_1\lambda_2 + \lambda_2 \lambda_3 + \lambda_3\lambda_1 }{(\lambda_1+\lambda_2+\lambda_3)^2}\]
static anomalous_exponent_(msd, time_lag)[source]

Calculates the diffusion anomalous exponent

\[\beta = \frac{ \partial \log{ \left( \langle x^2 \rangle \right)} }{ \partial (\log{(t)}) }\]
  • msd – mean square displacement

  • time_lag – time interval


diffusion nomalous exponent

static asymmetry_(eigenvalues)[source]

Takes the eigenvalues of the gyration radius tensor to estimate the asymmetry between axis.

\[a = - \log{ \left(1 - \frac{ ( \lambda_1 - \lambda_2)^2}{2 ( \lambda_1 + \lambda_2)^2} \right)}\]

eigenvalues – eigenvalues of the gyration radius tensor


asymmetry coefficient


Compute every feature for the trajectory saved in self._r.

Return features

return the values of the features as a string.

static confinement_probability_(r0, D, t)[source]

Estimate the probability of Brownian particle with diffusivity \(D\) being trapped in the interval \([-r0, +r0]\) after a period of time t.

\[P(r, D, t) = \int_{-r_0}^{r_0} p(r, D, t) \mathrm{d}r\]
  • r – position

  • D – diffusivity

  • t – time length

Return probability

probability of the particle being confined

static efficiency_(trajectory)[source]

Calculates the efficiency of the movement, a measure that is related to the straightness.

\[E = \frac{|\mathbf{x}_{N-1} - \mathbf{x}_{0}|^2 } { (N-1) \sum_{i=1}^{N-1} |\mathbf{x}_{i} - \mathbf{x}_{i-1}|^2 }\]
Return efficiency

trajectory efficiency.

static fractal_dimension_(trajectory)[source]

Estimates the fractal dimension of the trajectory

\[\frac{\log{(N)} }{ \log{(dNL^{-1})}}\]
Return fractal_dimension

returns the fractal dimension

static gaussianity_(trajectory)[source]

measure of how close to a gaussian distribution is the trajectory.

\[g(n) = \frac{ \langle r_n^4 \rangle }{2 \langle r_n^2 \rangle^2}\]
Return gaussianity

measure of similarity to a gaussian function


computes the generalised Green-Kubo’s diffusion constant

static gyration_radius_(trajectory)[source]

Calculates the gyration radius tensor of the trajectory

\[R_{mn} = \frac{1}{2N^2} \sum_{i=1}^N \sum_{j=1}^N \left( r_{m}^{(i)} - r_{m}^{(j)} \right)\left( r_{n}^{(i)} - r_{n}^{(j)} \right)\, ,\]

where \(N\) is the number of segments of the trajectory, \(\mathbf{r}_i\) is the \(i\)-th position vector along the trajectory, \(m\) and \(n\) assume the values of the corresponding coordinates along the directions \(x, y, z\).

Return gyration_radius


static kurtosis_(trajectory, eigenvector)[source]
We obtain the kurtosis by projecting each position of the trajectory along the main principal eigenvector of the radius of gyration tensor

\(r_i^p = \mathbf{r} \cdot \hat{e}_1\) and then calculating the quartic moment

\[K = \frac{1}{N} \sum_{i=1}^N \frac{ \left(r_i^p - \langle r^p \rangle \right)^4}{\sigma_{r^p}^4} \, ,\]

where \(\langle r^p \rangle\) is the mean position of the projected trajectory and \(\sigma_{r^p}^2\) is the variance. The kurtosis measures the peakiness of the distribution of points in the trajectory.

Return kurtosis


static msd_ensemble_averaged(trajectory)[source]

calculates the ensemble-averaged mean squared displacement

\[\langle \mathbf{r}^2 \rangle (t) = \frac{1}{N-1} \sum_{n=1}^N |\mathbf{x}_{n}-\mathbf{x}_0|^2\]

where \(N\) is the number of trajectories, \(\mathbf{r}_n(t)\) is the position of the trajectory \(n\) at time \(t\).

Return msd

ensemble-averaged msd

static msd_ratio_(msd_ta, n1, n2)[source]

Ratio of the ensemble averaged mean squared displacements.

\[\langle r^2 \rangle_{\tau_1, \tau_2} = \frac{\langle r^2 \rangle_{\tau_1 }} {\langle r^2 \rangle_{\tau_2 }} - \frac{\tau_1}{\tau_2}\]


\[\tau_1 < \tau_2\]
Return msd_ratio

static msd_time_averaged(trajectory, tau)[source]

calculates the time-averaged mean squared displacement

\[\langle \mathbf{r}_{\tau}^2 \rangle = \frac{1}{T-\tau} \sum_{t=1}^{N-\tau} |\mathbf{x}_{t+\tau} - \mathbf{x}_{\tau} |^2\]

where \(\tau\) is the time interval (time lag) between the two positions and :math:`T is total trajectory time length.

  • trajectory – trajectory array

  • tau – time lag, it can be a single value or an array

Return msd

time-averaged MSD

static straightness_(trajectory)[source]

Estimates how much straight is the trajectory

\[S = \frac{|\mathbf{x}_{N-1} -\mathbf{x}_0 |} { \sum_{i=1}^{N-1} |\mathbf{x}_i - \mathbf{x}_{i-1}|}\]
Return straightness

measure of linearity

Trajectory Generator

trajpy.traj_generator.anomalous_diffusion(n_steps, n_samples, time_step, alpha)[source]

Generates an ensemble of anomalous trajectories.

  • n_steps – total number of steps

  • n_samples – number of simulations

  • time_step – time step

  • alpha – anomalous exponent

Return x, y

time, array containing N_sample trajectories with Nsteps

trajpy.traj_generator.confined_diffusion(radius, n_steps, n_samples, dx, y0, D, dt)[source]

Generates trajectories under confinement.

  • radius – confinement radius

  • n_steps – number of displacements

  • n_samples – number of trajectories

  • dx – displacement

  • y0 – initial position

  • D – diffusion coefficient

  • dt – time step

Return x, y

time, array containing N_samples trajectories with N_steps

trajpy.traj_generator.normal_diffusion(n_steps, n_samples, dx, y0, D, dt)[source]

Generates an ensemble of normal diffusion trajectories.

  • n_steps – total steps

  • n_samples – number of trajectories

  • dx – maximum step length

  • y0 – starting position

  • D – diffusivity

  • dt – time step

Return x, y

time, array containing N_samples trajectories with N_steps

trajpy.traj_generator.normal_distribution(u, D, dt)[source]

This is the steplength probability density function for normal diffusion.

  • u – absolute distance travelled by the particle durint the time interval dt

  • D – diffusivity

  • dt – time interval

Return pdf

probability density function

trajpy.traj_generator.save_to_file(y, param, path)[source]

Saves the trajectories to a file.

  • y – trajectory array

  • param – a parameter that characterizes the kind of trajectory

  • path – path to the folder where the file will be saved

trajpy.traj_generator.superdiffusion(velocity, n_steps, n_samples, y0, dt)[source]

Generates direct diffusion trajectories. Combine pairwise with normal diffusion components.

  • velocity – constant velocity

  • n_steps – number of time steps

  • n_samples – number of trajectories

  • y0 – initial position

  • dt – time interval

Return x, y

time, array containing N_samples trajectories with N_steps

trajpy.traj_generator.weierstrass_mandelbrot(t, n_displacements, alpha)[source]

Calculates the weierstrass mandelbrot function

\[W(t) = \sum_{n=-\infty}^{\infty} \frac{\cos{(\phi_n )} - \cos{(\gamma^n t^* + \phi_n )} }{\gamma^{n\alpha/2}} \, .\]
  • t – time step

  • n_displacements – number of displacements

  • alpha – anomalous exponent


anomalous step