AntroPy is a Python package for computing entropy and fractal dimension measures of time-series. It is designed for speed (Numba JIT compilation) and ease of use, and works on both 1-D and N-D arrays. Typical use cases include feature extraction from physiological signals (e.g. EEG, ECG, EMG), and signal processing research.
| Function | Description |
|---|---|
ant.perm_entropy |
Permutation entropy — captures ordinal patterns in the signal. |
ant.spectral_entropy |
Spectral (power-spectrum) entropy via FFT or Welch method. |
ant.svd_entropy |
Singular value decomposition entropy of the time-delay embedding matrix. |
ant.app_entropy |
Approximate entropy (ApEn) — regularity measure sensitive to the length of the signal. |
ant.sample_entropy |
Sample entropy (SampEn) — less biased alternative to ApEn. |
ant.lziv_complexity |
Lempel-Ziv complexity for symbolic / binary sequences. |
ant.num_zerocross |
Number of zero-crossings. |
ant.hjorth_params |
Hjorth mobility and complexity parameters. |
| Function | Description |
|---|---|
ant.petrosian_fd |
Petrosian fractal dimension. |
ant.katz_fd |
Katz fractal dimension. |
ant.higuchi_fd |
Higuchi fractal dimension — slope of log curve-length vs log interval. |
ant.detrended_fluctuation |
Detrended fluctuation analysis (DFA) — estimates the Hurst / scaling exponent. |
AntroPy requires Python 3.10+ and depends on NumPy (≥ 1.22.4), SciPy (≥ 1.8.0), scikit-learn (≥ 1.2.0), and Numba (≥ 0.57).
# pip
pip install antropy
# uv
uv pip install antropy
# conda
conda install -c conda-forge antropygit clone https://github.com/raphaelvallat/antropy.git
cd antropy
uv pip install --group=test --editable .
pytest --verboseimport numpy as np
import antropy as ant
np.random.seed(1234567)
x = np.random.normal(size=3000)
print(ant.perm_entropy(x, normalize=True))
print(ant.spectral_entropy(x, sf=100, method='welch', normalize=True))
print(ant.svd_entropy(x, normalize=True))
print(ant.app_entropy(x))
print(ant.sample_entropy(x))
print(ant.hjorth_params(x)) # mobility in samples⁻¹
print(ant.hjorth_params(x, sf=100)) # mobility in Hz
print(ant.num_zerocross(x))
print(ant.lziv_complexity('01111000011001', normalize=True))0.9995 # perm_entropy (0 = regular, 1 = random) 0.9941 # spectral_entropy (0 = pure tone, 1 = white noise) 0.9999 # svd_entropy 2.0152 # app_entropy 2.1986 # sample_entropy (1.4313, 1.2153) # hjorth (mobility, complexity) (143.1339, 1.2153) # hjorth with sf=100 Hz 1531 # num_zerocross 1.3598 # lziv_complexity (normalized)
print(ant.petrosian_fd(x))
print(ant.katz_fd(x))
print(ant.higuchi_fd(x))
print(ant.detrended_fluctuation(x))1.0311 # petrosian_fd 5.9543 # katz_fd 2.0037 # higuchi_fd (≈ 2 for white noise) 0.4790 # DFA alpha (≈ 0.5 for white noise)
Most functions accept N-D arrays and an axis argument, making it easy to process
multi-channel data in a single call:
import numpy as np
import antropy as ant
# 4 channels × 3000 samples
X = np.random.normal(size=(4, 3000))
pe = ant.perm_entropy(X, normalize=True, axis=-1) # shape (4,)
mob, com = ant.hjorth_params(X, sf=256, axis=-1) # shape (4,) each
nzc = ant.num_zerocross(X, normalize=True, axis=-1) # shape (4,)
se = ant.spectral_entropy(X, sf=256, normalize=True) # shape (4,)Benchmarks on a 1000-sample signal (MacBook Pro M1 Max, 2021):
| Function | Time |
|---|---|
ant.perm_entropy |
53 µs |
ant.spectral_entropy |
113 µs |
ant.svd_entropy |
24 µs |
ant.app_entropy |
1.4 ms |
ant.sample_entropy |
910 µs |
ant.petrosian_fd |
6 µs |
ant.katz_fd |
9 µs |
ant.higuchi_fd |
7 µs |
ant.detrended_fluctuation |
100 µs |
Numba functions (sample_entropy, higuchi_fd, detrended_fluctuation) incur a one-time compilation cost on the first call.
AntroPy was created and is maintained by Raphael Vallat. Contributions are welcome — feel free to open an issue or submit a pull request on GitHub.
Note: this program is provided with NO WARRANTY OF ANY KIND. Always validate results against known references.
Several functions in AntroPy were adapted from:
- MNE-features — Jean-Baptiste Schiratti & Alexandre Gramfort
- pyEntropy — Nikolay Donets
- pyrem — Quentin Geissmann
- nolds — Christopher Scholzel
