Python DataMatrix

An intuitive, Pythonic way to work with tabular data.

Sebastiaan Mathôt
Copyright 2015-2023
https://pydatamatrix.eu/

About
Features
Ultra-short cheat sheet
Documentation
Dependencies
Installation
License

About

DataMatrix is an intuitive Python library for working with column-based, time-series, and multidimensional data. It's a light-weight and easy-to-use alternative to pandas.

DataMatrix is also one of the core libraries of OpenSesame, a graphical experiment builder for the social sciences, and Rapunzel, a modern code editor for numerical computing with Python and R.

Features

An intuitive syntax that makes your code easy to read
Mix tabular data with time series and multidimensional data in a single data structure
Support for large data by intelligent (and automatic) offloading of data to disk when memory is running low
Advanced memoization (caching)
Requires only the Python standard libraries (but you can use numpy to improve performance)
Compatible with your favorite data-science libraries:
- seaborn and matplotlib for plotting
- scipy, statsmodels, and pingouin for statistics
- mne for analysis of electroencephalographic (EEG) and magnetoencephalographic (MEG) data
- Convert to and from pandas.DataFrame
- Looks pretty inside a Jupyter Notebook

Ultra-short cheat sheet

from datamatrix import DataMatrix, io
# Read a DataMatrix from file
dm = io.readtxt('data.csv')
# Create a new DataMatrix
dm = DataMatrix(length=5)
# The first two rows
print(dm[:2])
# Create a new column and initialize it with the Fibonacci series
dm.fibonacci = 0, 1, 1, 2, 3
# You can also specify column names as if they are dict keys
dm['fibonacci'] = 0, 1, 1, 2, 3
# Remove 0 and 3 with a simple selection
dm = (dm.fibonacci > 0) & (dm.fibonacci < 3)
# Get a list of indices that match certain criteria
print(dm[(dm.fibonacci > 0) & (dm.fibonacci < 3)])
# Select 1, 1, and 2 by matching any of the values in a set
dm = dm.fibonacci == {1, 2}
# Select all odd numbers with a lambda expression
dm = dm.fibonacci == (lambda x: x % 2)
# Change all 1s to -1
dm.fibonacci[dm.fibonacci == 1] = -1
# The first two cells from the fibonacci column
print(dm.fibonacci[:2])
# Column mean
print(dm.fibonacci[...])
# Multiply all fibonacci cells by 2
dm.fibonacci_times_two = dm.fibonacci * 2
# Loop through all rows
for row in dm:
    print(row.fibonacci) # get the fibonacci cell from the row
# Loop through all columns
for colname, col in dm.columns:
    for cell in col: # Loop through all cells in the column
        print(cell) # do something with the cell
# Or just see which columns exist
print(dm.column_names)

Documentation

The basic documentation (including function and module references) is hosted on https://pydatamatrix.eu/. Additional tutorials can be found in the data-science course on https://pythontutorials.eu/.

Dependencies

DataMatrix requires only the Python standard library. That is, you can use it without installing any additional Python packages (although the pip and conda packages install some of the optional dependencies by default). Python 3.7 and higher are supported.

The following packages are required for extra functionality:

numpy and scipy for using the FloatColumn, IntColumn, SeriesColumn, MultiDimensionalColumn objects
pandas for conversion to and from pandas.DataFrame
mne for conversion to and from mne.Epochs and mne.TFR
fastnumbers for improved performance
prettytable for creating a text representation of a DataMatrix (e.g. to print it out)
openpyxl for reading and writing .xlsx files
json_tricks for hashing, serialization to and from json, and memoization (caching)
tomlkit for reading configuration from pyproject.toml
psutil for dynamic loading of large data

Installation

PyPi

pip install datamatrix

Historical note: The DataMatrix project used to correspond to another package of the same name, which was discontinued in 2010. If you want to install this package, you can do still do so by providing an explicit version (0.9 is the latest version of this package), as shown below. With thanks to dennogumi.org for handing over this project's entry on PyPi, thus avoiding much unnecessary confusion!

# Doesn't install datamatrix but a previous package by the same name!
pip install datamatrix==0.9

Anaconda

conda install datamatrix -c conda-forge

Ubuntu

sudo add-apt-repository ppa:smathot/cogscinl  # for stable releases
sudo add-apt-repository ppa:smathot/rapunzel  # for development releases
sudo apt-get update
sudo apt install python3-datamatrix

License

python-datamatrix is licensed under the GNU General Public License v3.

navidivan/datamatrix