uproot

uproot

ROOT I/O in pure Python and Numpy.

uproot (originally μproot, for “micro-Python ROOT”) is a reader and a writer of the ROOT file format using only Python and Numpy. Unlike the standard C++ ROOT implementation, uproot is only an I/O library, primarily intended to stream data into machine learning libraries in Python. Unlike PyROOT and root_numpy, uproot does not depend on C++ ROOT. Instead, it uses Numpy to cast blocks of data from the ROOT file as Numpy arrays.

Python does not necessarily mean slow. As long as the data blocks (“baskets”) are large, this “array at a time” approach can even be faster than “event at a time” C++. Below, the rate of reading data into arrays with uproot is shown to be faster than C++ ROOT (left) and root_numpy (right), as long as the baskets are tens of kilobytes or larger (for a variable number of muons per event in an ensemble of different physics samples; higher is better).

uproot is not maintained by the ROOT project team, so post bug reports here as GitHub issues, not on a ROOT forum. Thanks!

Installation

Install uproot like any other Python package:

pip install uproot                       # maybe with sudo or --user, or in virtualenv

or install with conda:

conda config --add channels conda-forge  # if you haven't added conda-forge already
conda install uproot

The pip installer automatically installs strict dependencies; the conda installer also installs optional dependencies (except for Pandas).

Strict dependencies:

Optional dependencies:

  • lz4 to read/write lz4-compressed ROOT files
  • xxhash to read/write lz4-compressed ROOT files
  • lzma to read/write lzma-compressed ROOT files in Python 2
  • xrootd to access remote files through XRootD
  • requests to access remote files through HTTP
  • pandas to fill Pandas DataFrames instead of Numpy arrays

Reminder: you do not need C++ ROOT to run uproot.

Testing and development requirements

The packages need to run the test suite can be installed via pip install "uproot[testing]". Here is an example how to set up a development

git clone https://github.com/scikit-hep/uproot.git  # alternatively your own fork
cd uproot
python -m venv venv  # create a virtual environment in the folder venv
. venv/bin/activate  # activate the Python environment
pip install -e ".[testing]"   # installs uproot in editable mode with all packages required for testing
pytest               # run the test suite

Tutorial

See the project homepage for a tutorial.

Run the tutorial on Binder.