numpy extension
Project description
NumPy Extensions
An extension library for NumPy that implements common array operations not present in NumPy.
npext.fill_na(...)
npext.drop_na(...)
npext.rolling(...)
npext.expanding(...)
npext.rolling_apply(...)
npext.expanding_apply(...)
# etc
Documentation
Installation
Regular installation:
pip install numpy_ext
For development:
git clone https://github.com/3jane/numpy_ext.git
cd numpy_ext
pip install -e .[dev] # note: make sure you are using pip>=20
Examples
Here are few common examples of how the library is used. The rest is available in the documentation.
- Apply a function to a rolling window over the provided array
import numpy as np
import numpy_ext as npext
a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
window = 3
npext.rolling_apply(np.sum, window, a)
> array([nan, nan, 3., 6., 9., 12., 15., 18., 21., 24.])
- Same as the above, but with a custom function, two input arrays and parallel computation using
joblib
:
def func(array_first, array_second, param):
return (np.min(array_first) + np.sum(array_second)) * param
a = np.array([0, 1, 2, 3])
b = np.array([3, 2, 1, 0])
npext.rolling_apply(func, 2, a, b, n_jobs=2, param=-1)
> array([nan, -5., -4., -3.])
- Same as the first example, but using rolling function:
a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
window = 3
rolls = npext.rolling(a, window, as_array=True)
np.sum(rolls, axis=1)
> array([nan, nan, 3., 6., 9., 12., 15., 18., 21., 24.])
License
The software is distributed under MIT license.
Project details
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
numpy_ext-0.9.1.tar.gz
(5.9 kB
view hashes)
Built Distribution
Close
Hashes for numpy_ext-0.9.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ef9f9ba549515e1791d6a9452d2d3fb9aa498b1b5b9f8f65247b182e27cd85ff |
|
MD5 | fea4181bac8da8911a4b5c2c6d3e1c6c |
|
BLAKE2b-256 | 49d9d969b9a3612df223f6b089217d20d1c539cb87126680a312b60aabc1efc8 |