A package providing an unit system for numpy multidimensionnal arrays.
Project description
numpy-unit
This package provides a tool for scientific computing by keeping track of the unit when performing classical operations on a multi-dimensionnal array with (almost) no extra-cost comparing to the standard numpy array.
The ArrayUnit
class supports every operation a numpy.ndarray can handle (because it is a derived class of numpy.ndarray) but the operators are overloaded in order to perform transformations on the Unit
contained in every ArrayUnit
.
Install
pip install numpy-unit
Examples
>>> import numpy as np
>>> from numpy_unit import Unit, ArrayUnit
>>>
>>> m = Unit('m')
>>> sm2 = Unit('s', -2)
>>> complex_unit = Unit({'my_unit': 0.5, '€': 1, 'capita': 2}) * (sm2**0.5) / m
>>> print(complex_unit)
€·capita²·my_unit^0.5·m⁻¹·s⁻¹
>>>
>>> arr = np.linspace(1,10,10, dtype=float)
>>> a = ArrayUnit(arr, m)
>>> b = ArrayUnit(arr**2, sm2)
>>> print(a, '\\n+\\n', 1, '\\n=\\n', a + 1)
[ 1. 2. 3. 4. 5. 6. 7. 8. 9. 10.] m
+
1
=
[ 2. 3. 4. 5. 6. 7. 8. 9. 10. 11.] m
>>> print(a, '\\n*\\n', b, '\\n=\\n', a * b)
[ 1. 2. 3. 4. 5. 6. 7. 8. 9. 10.] m
*
[ 1. 4. 9. 16. 25. 36. 49. 64. 81. 100.] s⁻²
=
[ 1. 8. 27. 64. 125. 216. 343. 512. 729. 1000.] m·s⁻²
The following rules applied (where {op} is one of the following: [+
, -
, *
, /
, //
, %
]):
- ArrayUnit {op} Object returns an ArrayUnit with the same unit as the ArrayUnit
- Object {op} ArrayUnit returns an ArrayUnit with the same unit as the ArrayUnit
- ArrayUnit {op} ArrayUnit returns an ArrayUnit combining the Unit of the 2 ArrayUnit or an Error
- An Error might be raised only when two ArrayUnit are conflicting and that ArrayUnit.is_strict is set to True. Otherwise, it would print a warning.
- An ArrayUnit is equal to a numpy.ndarray if and only if their underlying arrays are equal (np.array_equal) and the Unit of the ArrayUnit is empty.
Development
Doc of the master branch on readthedocs.io.
Features
- Basic unit system handling multiplication, division, modulo and power
- ArrayUnit wrapper for unit + ndarray
- Operators on ArrayUnit (and their variants r{op} and i{op})
- eq, ne
- add
- sub
- mul
- truediv, floordiv
- mod
- pow (but not rpow)
- Basic ndarray function
- mean
- std
- ...
- conda release
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
Built Distribution
Hashes for numpy_unit-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4ca911cc5b17d79c4f1a30aa128703a836c3149939279449ef2342da1dbfd184 |
|
MD5 | 2469512c960cfd50b2af130f1797c761 |
|
BLAKE2b-256 | a69626bfcd76332d4357d812bba047ad757640a1f6db2e922282199a2eac2923 |