| Name | numpy-groupies JSON |
| Version |
0.11.2
JSON |
| download |
| home_page | None |
| Summary | Optimised tools for group-indexing operations: aggregated sum and more. |
| upload_time | 2024-07-31 18:31:31 |
| maintainer | None |
| docs_url | None |
| author | None |
| requires_python | >=3.9 |
| license | Copyright (c) 2016, numpy-groupies developers All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| keywords |
accumarray
aggregate
groupby
grouping
indexing
|
| VCS |
 |
| bugtrack_url |
|
| requirements |
No requirements were recorded.
|
| Travis-CI |
No Travis.
|
| coveralls test coverage |
No coveralls.
|
[](https://github.com/ml31415/numpy-groupies/actions)
[](https://pypi.org/project/numpy-groupies/)
[](https://anaconda.org/conda-forge/numpy_groupies)
# numpy-groupies
This package consists of a small library of optimised tools for doing things that can roughly
be considered "group-indexing operations". The most prominent tool is `aggregate`, which is
described in detail further down the page.
## Installation
If you have `pip`, then simply:
```
pip install numpy_groupies
```
Note that `numpy_groupies` doesn't have any compulsory dependencies (even `numpy` is optional)
so you should be able to install it fairly easily even without a package manager. If you just
want one particular implementation of `aggregate` (e.g. `aggregate_numpy.py`), you can download
that one file, and copy-paste the contents of `utils.py` into the top of that file (replacing
the `from .utils import (...)` line).
## aggregate
```python
import numpy as np
import numpy_groupies as npg
group_idx = np.array([ 3, 0, 0, 1, 0, 3, 5, 5, 0, 4])
a = np.array([13.2, 3.5, 3.5,-8.2, 3.0,13.4,99.2,-7.1, 0.0,53.7])
npg.aggregate(group_idx, a, func='sum', fill_value=0)
# >>> array([10.0, -8.2, 0.0, 26.6, 53.7, 92.1])
```
`aggregate` takes an array of values, and an array giving the group number for each of those values.
It then returns the sum (or mean, or std, or any, ...etc.) of the values in each group. You have
probably come across this idea before - see [Matlab's `accumarray` function](http://uk.mathworks.com/help/matlab/ref/accumarray.html?refresh=true), or
[`pandas` groupby concept](http://pandas.pydata.org/pandas-docs/dev/groupby.html), or
[MapReduce paradigm](http://en.wikipedia.org/wiki/MapReduce), or simply the [basic histogram](https://en.wikipedia.org/wiki/Histogram).
A couple of implemented functions do not reduce the data, instead it calculates values cumulatively
while iterating over the data or permutates them. The output size matches the input size.
```python
group_idx = np.array([4, 3, 3, 4, 4, 1, 1, 1, 7, 8, 7, 4, 3, 3, 1, 1])
a = np.array([3, 4, 1, 3, 9, 9, 6, 7, 7, 0, 8, 2, 1, 8, 9, 8])
npg.aggregate(group_idx, a, func='cumsum')
# >>> array([3, 4, 5, 6,15, 9,15,22, 7, 0,15,17, 6,14,31,39])
```
### Inputs
The function accepts various different combinations of inputs, producing various different shapes of output.
We give a brief description of the general meaning of the inputs and then go over the different combinations
in more detail:
* `group_idx` - array of non-negative integers to be used as the "labels" with which to group the values in `a`.
* `a` - array of values to be aggregated.
* `func='sum'` - the function to use for aggregation. See the section below for more details.
* `size=None` - the shape of the output array. If `None`, the maximum value in `group_idx` will set the size of the output.
* `fill_value=0` - value to use for output groups that do not appear anywhere in the `group_idx` input array.
* `order='C'` - for multidimensional output, this controls the layout in memory, can be `'F'` for fortran-style.
* `dtype=None` - the`dtype` of the output. `None` means choose a sensible type for the given `a`, `func`, and `fill_value`.
* `axis=None` - explained below.
* `ddof=0` - passed through into calculations of variance and standard deviation (see section on functions).
* Form 1 is the simplest, taking `group_idx` and `a` of matching 1D lengths, and producing a 1D output.
* Form 2 is similar to Form 1, but takes a scalar `a`, which is broadcast out to the length of `group_idx`. Note that this is generally not that useful.
* Form 3 is more complicated. `group_idx` is the same length as the `a.shape[axis]`. The groups are broadcast out along the other axis/axes of `a`, thus the output is of shape `n_groups x a.shape[0] x ... x a.shape[axis-1] x a.shape[axis+1] x ... a.shape[-1]`, i.e. the output has two or more dimensions.
* Form 4 also produces output with two or more dimensions, but for very different reasons to Form 3. Here `a` is 1D and `group_idx` is exactly `2D`, whereas in Form 3 `a` is `ND`, `group_idx` is `1D`, and we provide a value for `axis`. The length of `a` must match `group_idx.shape[1]`, the value of `group_idx.shape[0]` determines the number of dimensions in the output, i.e. `group_idx[:,99]` gives the `(x,y,z)` group indices for the `a[99]`.
* Form 5 is the same as Form 4 but with scalar `a`. As with Form 2, this is rarely that helpful.
**Note on performance.** The `order` of the output is unlikely to affect performance of `aggregate` (although it may affect your downstream usage of that output), however the order of multidimensional `a` or `group_idx` can affect performance: in Form 4 it is best if columns are contiguous in memory within `group_idx`, i.e. `group_idx[:, 99]` corresponds to a contiguous chunk of memory; in Form 3 it's best if all the data in `a` for `group_idx[i]` is contiguous, e.g. if `axis=1` then we want `a[:, 55]` to be contiguous.
### Available functions
By default, `aggregate` assumes you want to sum the values within each group, however you can specify another
function using the `func` kwarg. This `func` can be any custom callable, however you will likely want one of
the following optimized functions. Note that not all functions might be provided by all implementations.
* `'sum'` - sum of items within each group (see example above).
* `'prod'` - product of items within each group
* `'mean'` - mean of items within each group
* `'var'`- variance of items within each group. Use `ddof` kwarg for degrees of freedom. The divisor used in calculations is `N - ddof`, where `N` represents the number of elements. By default `ddof` is zero.
* `'std'` - standard deviation of items within each group. Use `ddof` kwarg for degrees of freedom (see `var` above).
* `'min'` - minimum value of items within each group.
* `'max'` - maximum value of items within each group.
* `'first'` - first item in `a` from each group.
* `'last'` - last item in `a` from each group.
* `'argmax'` - the index in `a` of the maximum value in each group.
* `'argmin'` - the index in `a` of the minimum value in each group.
The above functions also have a `nan`-form, which skip the `nan` values instead of propagating them to the result of the calculation:
* `'nansum'`, `'nanprod'`, `'nanmean'`, `'nanvar'`, `'nanstd'`, `'nanmin'`, `'nanmax'`, `'nanfirst'`, `'nanlast'`, `'nanargmax'`, `'nanargmin'`
The following functions are slightly different in that they always return boolean values. Their treatment of nans is also different from above:
* `'all'` - `True` if all items within a group are truethy. Note that `np.all(nan)` is `True`, i.e. `nan` is actually truethy.
* `'any'` - `True` if any items within a group are truethy.
* `'allnan'` - `True` if all items within a group are `nan`.
* `'anynan'` - `True` if any items within a group are `nan`.
The following functions don't reduce the data, but instead produce an output matching the size of the input:
* `'cumsum'` - cumulative sum of items within each group.
* `'cumprod'` - cumulative product of items within each group. (numba only)
* `'cummin'` - cumulative minimum of items within each group. (numba only)
* `'cummax'` - cumulative maximum of items within each group. (numba only)
* `'sort'` - sort the items within each group in ascending order, use reverse=True to invert the order.
Finally, there are three functions which don't reduce each group to a single value, instead they return the full
set of items within the group:
* `'array'` - simply returns the grouped items, using the same order as appeared in `a`. (numpy only)
### Examples
Compute sums of consecutive integers, and then compute products of those consecutive integers.
```python
group_idx = np.arange(5).repeat(3)
# group_idx: array([0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4])
a = np.arange(group_idx.size)
# a: array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14])
x = npg.aggregate(group_idx, a) # sum is default
# x: array([ 3, 12, 21, 30, 39])
x = npg.aggregate(group_idx, a, 'prod')
# x: array([ 0, 60, 336, 990, 2184])
```
Get variance ignoring nans, setting all-nan groups to `nan`.
```python
x = npg.aggregate(group_idx, a, func='nanvar', fill_value=nan)
```
Count the number of elements in each group. Note that this is equivalent to doing `np.bincount(group_idx)`,
indeed that is how the numpy implementation does it.
```python
x = npg.aggregate(group_idx, 1)
```
Sum 1000 values into a three-dimensional cube of size 15x15x15. Note that in this example all three dimensions
have the same size, but that doesn't have to be the case.
```python
group_idx = np.random.randint(0, 15, size=(3, 1000))
a = np.random.random(group_idx.shape[1])
x = npg.aggregate(group_idx, a, func="sum", size=(15,15,15), order="F")
# x.shape: (15, 15, 15)
# np.isfortran(x): True
```
Use a custom function to generate some strings.
```python
group_idx = np.array([1, 0, 1, 4, 1])
a = np.array([12.0, 3.2, -15, 88, 12.9])
x = npg.aggregate(group_idx, a,
func=lambda g: ' or maybe '.join(str(gg) for gg in g), fill_value='')
# x: ['3.2', '12.0 or maybe -15.0 or maybe 12.9', '', '', '88.0']
```
Use the `axis` arg in order to do a sum-aggregation on three rows simultaneously.
```python
a = np.array([[99, 2, 11, 14, 20],
[33, 76, 12, 100, 71],
[67, 10, -8, 1, 9]])
group_idx = np.array([[3, 3, 7, 0, 0]])
x = npg.aggregate(group_idx, a, axis=1)
# x : [[ 34, 0, 0, 101, 0, 0, 0, 11],
# [171, 0, 0, 109, 0, 0, 0, 12],
# [ 10, 0, 0, 77, 0, 0, 0, -8]]
```
### Multiple implementations
There are multiple implementations of `aggregate` provided. If you use `from numpy_groupies import aggregate`,
the best available implementation will automatically be selected. Otherwise you can pick a specific version directly
like `from numpy_groupies import aggregate_nb as aggregate` or by importing aggregate from the implementing module
`from numpy_groupies.aggregate_weave import aggregate`.
Currently the following implementations exist:
* **numpy** - This is the default implementation. It uses plain `numpy`, mainly relying on `np.bincount` and basic indexing magic. It comes without other dependencies except `numpy` and shows reasonable performance for the occasional usage.
* **numba** - This is the most performant implementation, based on jit compilation provided by numba and LLVM.
* **pure python** - This implementation has no dependencies and uses only the standard library. It's horribly slow and should only be used, if there is no numpy available.
* **numpy ufunc** - *Only for benchmarking.* This implementation uses the `.at` method of numpy's `ufunc`s (e.g. `add.at`), which would appear to be designed for performing exactly the same calculation that `aggregate` executes, however the numpy implementation is rather incomplete.
* **pandas** - *Only for reference.* The pandas' `groupby` concept is the same as the task performed by `aggregate`. However, `pandas` is not actually faster than the default `numpy` implementation. Also, note that there may be room for improvement in the way that `pandas` is utilized here. Most notably, when computing multiple aggregations of the same data (e.g. `'min'` and `'max'`) pandas could potentially be used more efficiently.
All implementations have the same calling syntax and produce the same outputs, to within some floating-point error.
However some implementations only support a subset of the valid inputs and will sometimes throw `NotImplementedError`.
### Benchmarks
Scripts for testing and benchmarking are included in this repository. For benchmarking, run
`python -m numpy_groupies.benchmarks.generic` from the root of this repository.
Below we are using `500,000` indices uniformly picked from `[0, 1000)`. The values of `a` are uniformly picked from
the interval `[0,1)`, with anything less than `0.2` then set to 0 (in order to serve as falsy values in boolean operations).
For `nan-` operations another 20% of the values are set to nan, leaving the remainder on the interval `[0.2,0.8)`.
The benchmarking results are given in ms for an i7-7560U running at 2.40GHz:
| function | ufunc | numpy | numba | pandas |
|-----------|---------|---------|---------|---------|
| sum | 1.950 | 1.728 | 0.708 | 11.832 |
| prod | 2.279 | 2.349 | 0.709 | 11.649 |
| min | 2.472 | 2.489 | 0.716 | 11.686 |
| max | 2.457 | 2.480 | 0.745 | 11.598 |
| len | 1.481 | 1.270 | 0.635 | 10.932 |
| all | 37.186 | 3.054 | 0.892 | 12.587 |
| any | 35.278 | 5.157 | 0.890 | 12.845 |
| anynan | 5.783 | 2.126 | 0.762 | 144.740 |
| allnan | 7.971 | 4.367 | 0.774 | 144.507 |
| mean | ---- | 2.500 | 0.825 | 13.284 |
| std | ---- | 4.528 | 0.965 | 12.193 |
| var | ---- | 4.269 | 0.969 | 12.657 |
| first | ---- | 1.847 | 0.811 | 11.584 |
| last | ---- | 1.309 | 0.581 | 11.842 |
| argmax | ---- | 3.504 | 1.411 | 293.640 |
| argmin | ---- | 6.996 | 1.347 | 290.977 |
| nansum | ---- | 5.388 | 1.569 | 15.239 |
| nanprod | ---- | 5.707 | 1.546 | 15.004 |
| nanmin | ---- | 5.831 | 1.700 | 14.292 |
| nanmax | ---- | 5.847 | 1.731 | 14.927 |
| nanlen | ---- | 3.170 | 1.529 | 14.529 |
| nanall | ---- | 6.499 | 1.640 | 15.931 |
| nanany | ---- | 8.041 | 1.656 | 15.839 |
| nanmean | ---- | 5.636 | 1.583 | 15.185 |
| nanvar | ---- | 7.514 | 1.682 | 15.643 |
| nanstd | ---- | 7.292 | 1.666 | 15.104 |
| nanfirst | ---- | 5.318 | 2.096 | 14.432 |
| nanlast | ---- | 4.943 | 1.473 | 14.637 |
| nanargmin | ---- | 7.977 | 1.779 | 298.911 |
| nanargmax | ---- | 5.869 | 1.802 | 301.022 |
| cumsum | ---- | 71.713 | 1.119 | 8.864 |
| cumprod | ---- | ---- | 1.123 | 12.100 |
| cummax | ---- | ---- | 1.062 | 12.133 |
| cummin | ---- | ---- | 0.973 | 11.908 |
| arbitrary | ---- | 147.853 | 46.690 | 129.779 |
| sort | ---- | 167.699 | ---- | ---- |
_Linux(x86_64), Python 3.10.12, Numpy 1.25.2, Numba 0.58.0, Pandas 2.0.2_
## Development
This project was started by @ml31415 and the `numba` and `weave` implementations are by him. The pure
python and `numpy` implementations were written by @d1manson.
The authors hope that `numpy`'s `ufunc.at` methods or some other implementation of `aggregate` within
`numpy` or `scipy` will eventually be fast enough, to make this package redundant. Numpy 1.25 actually
contained major [improvements on ufunc speed](https://numpy.org/doc/stable/release/1.25.0-notes.html),
which reduced the speed gap between numpy and the numba implementation a lot.
Raw data
{
"_id": null,
"home_page": null,
"name": "numpy-groupies",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.9",
"maintainer_email": "Deepak Cherian <dcherian@ucar.edu>",
"keywords": "accumarray, aggregate, groupby, grouping, indexing",
"author": null,
"author_email": "Michael L\u00f6ffler <ml@occam.com.ua>, Daniel Manson <danielmanson.uk@gmail.com>",
"download_url": "https://files.pythonhosted.org/packages/26/56/4fdd73388ec13a1ac3d4b27856a67b098eac6c120bbb43b42b2822e20f6c/numpy_groupies-0.11.2.tar.gz",
"platform": null,
"description": "[](https://github.com/ml31415/numpy-groupies/actions)\n[](https://pypi.org/project/numpy-groupies/)\n[](https://anaconda.org/conda-forge/numpy_groupies)\n\n\n\n# numpy-groupies\n\nThis package consists of a small library of optimised tools for doing things that can roughly \nbe considered \"group-indexing operations\". The most prominent tool is `aggregate`, which is \ndescribed in detail further down the page.\n\n\n## Installation\nIf you have `pip`, then simply:\n```\npip install numpy_groupies\n```\nNote that `numpy_groupies` doesn't have any compulsory dependencies (even `numpy` is optional) \nso you should be able to install it fairly easily even without a package manager. If you just \nwant one particular implementation of `aggregate` (e.g. `aggregate_numpy.py`), you can download \nthat one file, and copy-paste the contents of `utils.py` into the top of that file (replacing \nthe `from .utils import (...)` line).\n\n\n## aggregate\n\n\n```python\nimport numpy as np\nimport numpy_groupies as npg\ngroup_idx = np.array([ 3, 0, 0, 1, 0, 3, 5, 5, 0, 4])\na = np.array([13.2, 3.5, 3.5,-8.2, 3.0,13.4,99.2,-7.1, 0.0,53.7])\nnpg.aggregate(group_idx, a, func='sum', fill_value=0)\n# >>> array([10.0, -8.2, 0.0, 26.6, 53.7, 92.1])\n```\n`aggregate` takes an array of values, and an array giving the group number for each of those values. \nIt then returns the sum (or mean, or std, or any, ...etc.) of the values in each group. You have \nprobably come across this idea before - see [Matlab's `accumarray` function](http://uk.mathworks.com/help/matlab/ref/accumarray.html?refresh=true), or\n [`pandas` groupby concept](http://pandas.pydata.org/pandas-docs/dev/groupby.html), or\n [MapReduce paradigm](http://en.wikipedia.org/wiki/MapReduce), or simply the [basic histogram](https://en.wikipedia.org/wiki/Histogram).\n\nA couple of implemented functions do not reduce the data, instead it calculates values cumulatively\nwhile iterating over the data or permutates them. The output size matches the input size.\n\n```python\ngroup_idx = np.array([4, 3, 3, 4, 4, 1, 1, 1, 7, 8, 7, 4, 3, 3, 1, 1])\na = np.array([3, 4, 1, 3, 9, 9, 6, 7, 7, 0, 8, 2, 1, 8, 9, 8])\nnpg.aggregate(group_idx, a, func='cumsum')\n# >>> array([3, 4, 5, 6,15, 9,15,22, 7, 0,15,17, 6,14,31,39])\n```\n\n\n### Inputs\nThe function accepts various different combinations of inputs, producing various different shapes of output. \nWe give a brief description of the general meaning of the inputs and then go over the different combinations \nin more detail:\n\n* `group_idx` - array of non-negative integers to be used as the \"labels\" with which to group the values in `a`.\n* `a` - array of values to be aggregated.\n* `func='sum'` - the function to use for aggregation. See the section below for more details.\n* `size=None` - the shape of the output array. If `None`, the maximum value in `group_idx` will set the size of the output.\n* `fill_value=0` - value to use for output groups that do not appear anywhere in the `group_idx` input array.\n* `order='C'` - for multidimensional output, this controls the layout in memory, can be `'F'` for fortran-style.\n* `dtype=None` - the`dtype` of the output. `None` means choose a sensible type for the given `a`, `func`, and `fill_value`.\n* `axis=None` - explained below.\n* `ddof=0` - passed through into calculations of variance and standard deviation (see section on functions).\n\n\n\n* Form 1 is the simplest, taking `group_idx` and `a` of matching 1D lengths, and producing a 1D output.\n* Form 2 is similar to Form 1, but takes a scalar `a`, which is broadcast out to the length of `group_idx`. Note that this is generally not that useful.\n* Form 3 is more complicated. `group_idx` is the same length as the `a.shape[axis]`. The groups are broadcast out along the other axis/axes of `a`, thus the output is of shape `n_groups x a.shape[0] x ... x a.shape[axis-1] x a.shape[axis+1] x ... a.shape[-1]`, i.e. the output has two or more dimensions.\n* Form 4 also produces output with two or more dimensions, but for very different reasons to Form 3. Here `a` is 1D and `group_idx` is exactly `2D`, whereas in Form 3 `a` is `ND`, `group_idx` is `1D`, and we provide a value for `axis`. The length of `a` must match `group_idx.shape[1]`, the value of `group_idx.shape[0]` determines the number of dimensions in the output, i.e. `group_idx[:,99]` gives the `(x,y,z)` group indices for the `a[99]`.\n* Form 5 is the same as Form 4 but with scalar `a`. As with Form 2, this is rarely that helpful.\n\n**Note on performance.** The `order` of the output is unlikely to affect performance of `aggregate` (although it may affect your downstream usage of that output), however the order of multidimensional `a` or `group_idx` can affect performance: in Form 4 it is best if columns are contiguous in memory within `group_idx`, i.e. `group_idx[:, 99]` corresponds to a contiguous chunk of memory; in Form 3 it's best if all the data in `a` for `group_idx[i]` is contiguous, e.g. if `axis=1` then we want `a[:, 55]` to be contiguous.\n\n\n### Available functions\nBy default, `aggregate` assumes you want to sum the values within each group, however you can specify another \nfunction using the `func` kwarg. This `func` can be any custom callable, however you will likely want one of\nthe following optimized functions. Note that not all functions might be provided by all implementations.\n\n* `'sum'` - sum of items within each group (see example above).\n* `'prod'` - product of items within each group\n* `'mean'` - mean of items within each group\n* `'var'`- variance of items within each group. Use `ddof` kwarg for degrees of freedom. The divisor used in calculations is `N - ddof`, where `N` represents the number of elements. By default `ddof` is zero.\n* `'std'` - standard deviation of items within each group. Use `ddof` kwarg for degrees of freedom (see `var` above).\n* `'min'` - minimum value of items within each group.\n* `'max'` - maximum value of items within each group.\n* `'first'` - first item in `a` from each group.\n* `'last'` - last item in `a` from each group.\n* `'argmax'` - the index in `a` of the maximum value in each group.\n* `'argmin'` - the index in `a` of the minimum value in each group.\n\nThe above functions also have a `nan`-form, which skip the `nan` values instead of propagating them to the result of the calculation:\n* `'nansum'`, `'nanprod'`, `'nanmean'`, `'nanvar'`, `'nanstd'`, `'nanmin'`, `'nanmax'`, `'nanfirst'`, `'nanlast'`, `'nanargmax'`, `'nanargmin'`\n\nThe following functions are slightly different in that they always return boolean values. Their treatment of nans is also different from above:\n* `'all'` - `True` if all items within a group are truethy. Note that `np.all(nan)` is `True`, i.e. `nan` is actually truethy.\n* `'any'` - `True` if any items within a group are truethy.\n* `'allnan'` - `True` if all items within a group are `nan`.\n* `'anynan'` - `True` if any items within a group are `nan`.\n\nThe following functions don't reduce the data, but instead produce an output matching the size of the input:\n* `'cumsum'` - cumulative sum of items within each group.\n* `'cumprod'` - cumulative product of items within each group. (numba only)\n* `'cummin'` - cumulative minimum of items within each group. (numba only)\n* `'cummax'` - cumulative maximum of items within each group. (numba only)\n* `'sort'` - sort the items within each group in ascending order, use reverse=True to invert the order.\n\nFinally, there are three functions which don't reduce each group to a single value, instead they return the full \nset of items within the group:\n* `'array'` - simply returns the grouped items, using the same order as appeared in `a`. (numpy only)\n\n\n### Examples\nCompute sums of consecutive integers, and then compute products of those consecutive integers.\n```python\ngroup_idx = np.arange(5).repeat(3)\n# group_idx: array([0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4])\na = np.arange(group_idx.size)\n# a: array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14])\nx = npg.aggregate(group_idx, a) # sum is default\n# x: array([ 3, 12, 21, 30, 39])\nx = npg.aggregate(group_idx, a, 'prod')\n# x: array([ 0, 60, 336, 990, 2184])\n```\n\nGet variance ignoring nans, setting all-nan groups to `nan`.\n```python\nx = npg.aggregate(group_idx, a, func='nanvar', fill_value=nan)\n```\n\nCount the number of elements in each group. Note that this is equivalent to doing `np.bincount(group_idx)`, \nindeed that is how the numpy implementation does it.\n```python\nx = npg.aggregate(group_idx, 1)\n```\n\nSum 1000 values into a three-dimensional cube of size 15x15x15. Note that in this example all three dimensions \nhave the same size, but that doesn't have to be the case.\n```python\ngroup_idx = np.random.randint(0, 15, size=(3, 1000))\na = np.random.random(group_idx.shape[1])\nx = npg.aggregate(group_idx, a, func=\"sum\", size=(15,15,15), order=\"F\")\n# x.shape: (15, 15, 15)\n# np.isfortran(x): True\n```\n\nUse a custom function to generate some strings.\n```python\ngroup_idx = np.array([1, 0, 1, 4, 1])\na = np.array([12.0, 3.2, -15, 88, 12.9])\nx = npg.aggregate(group_idx, a,\n func=lambda g: ' or maybe '.join(str(gg) for gg in g), fill_value='')\n# x: ['3.2', '12.0 or maybe -15.0 or maybe 12.9', '', '', '88.0']\n```\n\nUse the `axis` arg in order to do a sum-aggregation on three rows simultaneously.\n```python\na = np.array([[99, 2, 11, 14, 20],\n\t \t [33, 76, 12, 100, 71],\n\t\t [67, 10, -8, 1, 9]])\ngroup_idx = np.array([[3, 3, 7, 0, 0]])\nx = npg.aggregate(group_idx, a, axis=1)\n# x : [[ 34, 0, 0, 101, 0, 0, 0, 11],\n# [171, 0, 0, 109, 0, 0, 0, 12],\n# [ 10, 0, 0, 77, 0, 0, 0, -8]]\n```\n\n\n### Multiple implementations\nThere are multiple implementations of `aggregate` provided. If you use `from numpy_groupies import aggregate`, \nthe best available implementation will automatically be selected. Otherwise you can pick a specific version directly \nlike `from numpy_groupies import aggregate_nb as aggregate` or by importing aggregate from the implementing module \n`from numpy_groupies.aggregate_weave import aggregate`.\n\nCurrently the following implementations exist:\n* **numpy** - This is the default implementation. It uses plain `numpy`, mainly relying on `np.bincount` and basic indexing magic. It comes without other dependencies except `numpy` and shows reasonable performance for the occasional usage.\n* **numba** - This is the most performant implementation, based on jit compilation provided by numba and LLVM.\n* **pure python** - This implementation has no dependencies and uses only the standard library. It's horribly slow and should only be used, if there is no numpy available.\n* **numpy ufunc** - *Only for benchmarking.* This implementation uses the `.at` method of numpy's `ufunc`s (e.g. `add.at`), which would appear to be designed for performing exactly the same calculation that `aggregate` executes, however the numpy implementation is rather incomplete.\n* **pandas** - *Only for reference.* The pandas' `groupby` concept is the same as the task performed by `aggregate`. However, `pandas` is not actually faster than the default `numpy` implementation. Also, note that there may be room for improvement in the way that `pandas` is utilized here. Most notably, when computing multiple aggregations of the same data (e.g. `'min'` and `'max'`) pandas could potentially be used more efficiently.\n\nAll implementations have the same calling syntax and produce the same outputs, to within some floating-point error. \nHowever some implementations only support a subset of the valid inputs and will sometimes throw `NotImplementedError`.\n\n\n### Benchmarks\nScripts for testing and benchmarking are included in this repository. For benchmarking, run \n`python -m numpy_groupies.benchmarks.generic` from the root of this repository.\n\nBelow we are using `500,000` indices uniformly picked from `[0, 1000)`. The values of `a` are uniformly picked from \nthe interval `[0,1)`, with anything less than `0.2` then set to 0 (in order to serve as falsy values in boolean operations). \nFor `nan-` operations another 20% of the values are set to nan, leaving the remainder on the interval `[0.2,0.8)`.\n\nThe benchmarking results are given in ms for an i7-7560U running at 2.40GHz:\n\n| function | ufunc | numpy | numba | pandas |\n|-----------|---------|---------|---------|---------|\n| sum | 1.950 | 1.728 | 0.708 | 11.832 |\n| prod | 2.279 | 2.349 | 0.709 | 11.649 |\n| min | 2.472 | 2.489 | 0.716 | 11.686 |\n| max | 2.457 | 2.480 | 0.745 | 11.598 |\n| len | 1.481 | 1.270 | 0.635 | 10.932 |\n| all | 37.186 | 3.054 | 0.892 | 12.587 |\n| any | 35.278 | 5.157 | 0.890 | 12.845 |\n| anynan | 5.783 | 2.126 | 0.762 | 144.740 |\n| allnan | 7.971 | 4.367 | 0.774 | 144.507 |\n| mean | ---- | 2.500 | 0.825 | 13.284 |\n| std | ---- | 4.528 | 0.965 | 12.193 |\n| var | ---- | 4.269 | 0.969 | 12.657 |\n| first | ---- | 1.847 | 0.811 | 11.584 |\n| last | ---- | 1.309 | 0.581 | 11.842 |\n| argmax | ---- | 3.504 | 1.411 | 293.640 |\n| argmin | ---- | 6.996 | 1.347 | 290.977 |\n| nansum | ---- | 5.388 | 1.569 | 15.239 |\n| nanprod | ---- | 5.707 | 1.546 | 15.004 |\n| nanmin | ---- | 5.831 | 1.700 | 14.292 |\n| nanmax | ---- | 5.847 | 1.731 | 14.927 |\n| nanlen | ---- | 3.170 | 1.529 | 14.529 |\n| nanall | ---- | 6.499 | 1.640 | 15.931 |\n| nanany | ---- | 8.041 | 1.656 | 15.839 |\n| nanmean | ---- | 5.636 | 1.583 | 15.185 |\n| nanvar | ---- | 7.514 | 1.682 | 15.643 |\n| nanstd | ---- | 7.292 | 1.666 | 15.104 |\n| nanfirst | ---- | 5.318 | 2.096 | 14.432 |\n| nanlast | ---- | 4.943 | 1.473 | 14.637 |\n| nanargmin | ---- | 7.977 | 1.779 | 298.911 |\n| nanargmax | ---- | 5.869 | 1.802 | 301.022 |\n| cumsum | ---- | 71.713 | 1.119 | 8.864 |\n| cumprod | ---- | ---- | 1.123 | 12.100 |\n| cummax | ---- | ---- | 1.062 | 12.133 |\n| cummin | ---- | ---- | 0.973 | 11.908 |\n| arbitrary | ---- | 147.853 | 46.690 | 129.779 |\n| sort | ---- | 167.699 | ---- | ---- |\n\n_Linux(x86_64), Python 3.10.12, Numpy 1.25.2, Numba 0.58.0, Pandas 2.0.2_\n\n## Development\nThis project was started by @ml31415 and the `numba` and `weave` implementations are by him. The pure \npython and `numpy` implementations were written by @d1manson.\n\nThe authors hope that `numpy`'s `ufunc.at` methods or some other implementation of `aggregate` within\n`numpy` or `scipy` will eventually be fast enough, to make this package redundant. Numpy 1.25 actually\ncontained major [improvements on ufunc speed](https://numpy.org/doc/stable/release/1.25.0-notes.html), \nwhich reduced the speed gap between numpy and the numba implementation a lot.\n",
"bugtrack_url": null,
"license": "Copyright (c) 2016, numpy-groupies developers All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ",
"summary": "Optimised tools for group-indexing operations: aggregated sum and more.",
"version": "0.11.2",
"project_urls": {
"source": "https://github.com/ml31415/numpy-groupies",
"tracker": "https://github.com/ml31415/numpy-groupies/issues"
},
"split_keywords": [
"accumarray",
" aggregate",
" groupby",
" grouping",
" indexing"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "a442230a3fc58aa50ce28649f3f9162d83bbf3d77e29fdd397a4766d8a88409f",
"md5": "361955ab25affbcc65bd7a08bd920dea",
"sha256": "8d0d686160f860c12d97f12f457116f8b8bebfa33a68b8efcd24dcfedc2837f1"
},
"downloads": -1,
"filename": "numpy_groupies-0.11.2-py3-none-any.whl",
"has_sig": false,
"md5_digest": "361955ab25affbcc65bd7a08bd920dea",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.9",
"size": 40634,
"upload_time": "2024-07-31T18:31:29",
"upload_time_iso_8601": "2024-07-31T18:31:29.679588Z",
"url": "https://files.pythonhosted.org/packages/a4/42/230a3fc58aa50ce28649f3f9162d83bbf3d77e29fdd397a4766d8a88409f/numpy_groupies-0.11.2-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "26564fdd73388ec13a1ac3d4b27856a67b098eac6c120bbb43b42b2822e20f6c",
"md5": "3f61384baf84879c19621c6f7449dde0",
"sha256": "2fda978c4d28d2f1633a63972f425d0a7f2f12a75505d215b41b6de712e2ec4b"
},
"downloads": -1,
"filename": "numpy_groupies-0.11.2.tar.gz",
"has_sig": false,
"md5_digest": "3f61384baf84879c19621c6f7449dde0",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.9",
"size": 159015,
"upload_time": "2024-07-31T18:31:31",
"upload_time_iso_8601": "2024-07-31T18:31:31.321760Z",
"url": "https://files.pythonhosted.org/packages/26/56/4fdd73388ec13a1ac3d4b27856a67b098eac6c120bbb43b42b2822e20f6c/numpy_groupies-0.11.2.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-07-31 18:31:31",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "ml31415",
"github_project": "numpy-groupies",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "numpy-groupies"
}
