GroupBy operations for dask.array
Project description
flox
This project explores strategies for fast GroupBy reductions with dask.array. It used to be called dask_groupby
It was motivated by
(See a presentation about this package, from the Pangeo Showcase).
Acknowledgements
This work was funded in part by NASA-ACCESS 80NSSC18M0156 "Community tools for analysis of NASA Earth Observing System Data in the Cloud" (PI J. Hamman), and NCAR's Earth System Data Science Initiative. It was motivated by very very many discussions in the Pangeo community.
API
There are two main functions
flox.groupby_reduce(dask_array, by_dask_array, "mean")"pure" dask array interfaceflox.xarray.xarray_reduce(xarray_object, by_dataarray, "mean")"pure" xarray interface; though work is ongoing to integrate this package in xarray.
Implementation
See the documentation for details on the implementation.
Custom reductions
flox implements all common reductions provided by numpy_groupies in aggregations.py.
It also allows you to specify a custom Aggregation (again inspired by dask.dataframe),
though this might not be fully functional at the moment. See aggregations.py for examples.
mean = Aggregation(
# name used for dask tasks
name="mean",
# operation to use for pure-numpy inputs
numpy="mean",
# blockwise reduction
chunk=("sum", "count"),
# combine intermediate results: sum the sums, sum the counts
combine=("sum", "sum"),
# generate final result as sum / count
finalize=lambda sum_, count: sum_ / count,
# Used when "reindexing" at combine-time
fill_value=0,
# Used when any member of `expected_groups` is not found
final_fill_value=np.nan,
)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file flox-0.5.1.tar.gz.
File metadata
- Download URL: flox-0.5.1.tar.gz
- Upload date:
- Size: 308.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.10.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | ccadb7b32fec2046c154ae5623b3e646d155077e57b23882619205c7d0588cda |
|
| MD5 | fe29bebfa40eefebe7649cc62d6e2330 |
|
| BLAKE2b-256 | 8e5daceec44de085279e1df39b4a05e543c23522557cdbaabf395a8aae45c1a8 |
File details
Details for the file flox-0.5.1-py3-none-any.whl.
File metadata
- Download URL: flox-0.5.1-py3-none-any.whl
- Upload date:
- Size: 54.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.10.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | b11902d4ae2c5343085c4720654bea9b10468a5554c0c41702a617fbbc1a054d |
|
| MD5 | e617a55a7d9356b4d5f3dab24ed6fa13 |
|
| BLAKE2b-256 | 34326aa5518a12772d4df20bf4d847fce73815564b276070e1c526cffc5c5072 |
