Skip to main content

Common analysis utilities

Project description

anaties

Anaties (contraction of 'analysis utilities'). A place for common operations like signal smoothing that are useful across all my data analysis projects.

Installation and usage

Install with pip:

pip install anaties

When a new release is made, upgrade with:

pip install anaties --upgrade

Usage is simple. In your code:

import anaties as ana
ana.function_name()

You can test it out with:

import anaties as ana
print(ana.datetime_string())

plt.plot([0, 1], [0,1], color='k', linewidth=0.6)
ana.rect_highlight([0.25, 0.5])

All other functions are listed below.

Brief summary of all utilities

    signals.py (for 1d data arrays, or arrays of such arrays)
        - smooth: smooth a signal with a window (gaussian, etc)
        - smooth_rows: smooth each row of a 2d array using smooth()
        - power_spec: get the power spectral density or power spectrum
        - spectrogram: calculate/plot spectrogram of a signal
        - notch_filter: notch filter to attenuate specific frequency (e.g. 60hz)
        - bandpass_filter: allow through frequencies within low- and high-cutoff

    plots.py (basic plotting)
        - error_shade: plot line with shaded error region
        - freqhist: calculate/plot a relative frequency histogram
        - paired_bar: bar plot for paired data
        - plot_with_events: plot with vertical lines to indicate events
        - rect_highlight: overlay rectangular highlight to figure
        - vlines: add vertical lines to figure

    stats (basic statistical things)
        - collective_correlation: collective correlation coefficient
        - med_semed: median and std error of median of an array
        - mean_sem: mean and std error of the mean of an array
        - mean_std: mean and standard deviation of an array
        - se_mean: std err of mean of array
        - se_median: std error of median of array
        - cramers_v: cramers v for effect size for chi-square test

    helpers.py (generic utility functions for use everywhere)
        - datetime_string : return date_time string to use for naming files etc
        - file_exists: check to see if file exists
        - get_bins: get bin edges and centers, given limits and bin width
        - get_offdiag_vals: get lower off-diagonal values of a symmetric matrix
        - ind_limits: return indices that contain array data within range
        - is_symmetric: check if 2d array is symmetric
        - rand_rgb: returns random array of rgb values

Acknowledgments

To do: More important

  • finish adding tests.
  • use median instead of mean in spectrogram
  • add proper documentation and tests to stats module.
  • integrate vlines into pypi and version up (maybe good test for ci)
  • add ax return for all plot functions, when possible.
  • finish plots.twinx and make sure it works
  • add test for plots.error_shade.
  • Add return object for plots.rect_highlight()
  • consider adding directory_exists to helpers
  • paired_bar and mean_sem/std need to handle one point better (throws warning)
  • Add a proper suptitle fix in aplots it is a pita to add manually/remember: f.suptitle(..., fontsize=16) f.tight_layout() f.subplots_adjust(top=0.9)
  • For freqhist should I guarantee it sums to 1 even when bin widths don't match data limits? Probably not. Something to think about though.
  • In smoother, consider switching from filtfilt() to sosfiltfilt() for reasons laid out here: https://dsp.stackexchange.com/a/17255/51564
  • Convert notch filter to sos?
  • For spectral density estimation consider adding multitaper option. Good discussions: https://github.com/cokelaer/spectrum https://pyspectrum.readthedocs.io/en/latest/ https://mark-kramer.github.io/Case-Studies-Python/04.html
  • add ability to control event colors in spectrogram.
  • ind_limits: add checks for data, data_limits, clarify description and docs
  • Add numerical tests with random seed set not just graphical eyeball tests.

To do: longer term

  • Add audio playback of signals (see notes in audio_playback_workspace), incorporate this into some tests of filtering, etc.. simpleaudio package is too simple I think.
  • autodocs (sphinx?)
  • CI/CD with github actions
  • consider adding wavelets.
  • Add 3d array support for stat functions like mn_sem

Useful sources

Smoothing

What about wavelets?

I may add wavelets at some point, but it isn't plug-and-play enough for this repo. If you want to get started with wavelets in Python, I recommend http://ataspinar.com/2018/12/21/a-guide-for-using-the-wavelet-transform-in-machine-learning/

Tolerance values

For a discussion of the difference between relative and absolute tolerance values when testing floats for equality (for instance as used in helpers.is_symmetric()) see: https://stackoverflow.com/questions/65909842/what-is-rtol-for-in-numpys-allclose-function

Suggestions?

If there is something you'd like to see, please open an issue.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

anaties-0.1.4.tar.gz (21.1 kB view details)

Uploaded Source

Built Distribution

anaties-0.1.4-py3-none-any.whl (21.7 kB view details)

Uploaded Python 3

File details

Details for the file anaties-0.1.4.tar.gz.

File metadata

  • Download URL: anaties-0.1.4.tar.gz
  • Upload date:
  • Size: 21.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.3.0 pkginfo/1.7.1 requests/2.26.0 setuptools/58.2.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.7.6

File hashes

Hashes for anaties-0.1.4.tar.gz
Algorithm Hash digest
SHA256 02b393415de73ffa6d34ff660561ee0f181e6535051ce81eb3c7da2291d9a9d9
MD5 1bc5e4f5ec4eab84b15253c88b2dae9d
BLAKE2b-256 1733a8c2b7cd4435fb37cea3726bf648157463b2de2f438773b64f07d47a8ab8

See more details on using hashes here.

File details

Details for the file anaties-0.1.4-py3-none-any.whl.

File metadata

  • Download URL: anaties-0.1.4-py3-none-any.whl
  • Upload date:
  • Size: 21.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.3.0 pkginfo/1.7.1 requests/2.26.0 setuptools/58.2.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.7.6

File hashes

Hashes for anaties-0.1.4-py3-none-any.whl
Algorithm Hash digest
SHA256 c0531bbdbe637e800f41c255f01bf5ba396b364f3d8738d1de34d68176235359
MD5 6d7a9fa29b391e85faf227d0b212c39a
BLAKE2b-256 0064c9b94c054dacb7a0dc24913de6c70fbd99b1a26c65a373a9f7a6343d62d7

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page