Skip to main content

Polynomials as a numpy datatype

Project description

doc/.static/numpoly_logo.svg

circleci codecov pypi readthedocs

Numpoly is a generic library for creating, manipulating and evaluating arrays of polynomials based on numpy.ndarray objects.

Feature Overview

  • Intuitive interface for users experienced with numpy, as the library provides a high level of compatibility with the numpy.ndarray, including fancy indexing, broadcasting, numpy.dtype, vectorized operations to name a few.

  • Computationally fast evaluations of lots of functionality inherent from numpy.

  • Vectorized polynomial evaluation.

  • Support for arbitrary number of dimensions and name for the indeterminants.

  • Native support for lots of numpy.<name> functions using numpy’s compatibility layer (which also exists as numpoly.<name> equivalents).

  • Support for polynomial division through the operators /, % and divmod.

  • Extra polynomial specific attributes exposed on the polynomial objects like poly.exponents, poly.coefficients, poly.indeterminants etc.

  • Polynomial derivation through functions like numpoly.diff, numpoly.gradient, numpoly.hessian etc.

  • Decompose polynomial sums into vector of addends using numpoly.decompose.

  • Variable substitution through numpoly.call.

numpoly is currently being used as the backend is the uncertainty quantification library chaospy.

Installation

Installation should be straight forward:

pip install numpoly

And you should be ready to go. That is it. You should now be able to import the library in your Python REPL:

>>> import numpoly

Example Usage

Constructing polynomial is typically done using one of the available constructors:

>>> numpoly.monomial(start=0, stop=3, names=2)
polynomial([1, q0, q0**2, q1, q0*q1, q1**2])

It is also possible to construct your own from symbols:

>>> q0, q1 = numpoly.variable(2)
>>> numpoly.polynomial([1, q0**2-1, q0*q1, q1**2-1])
polynomial([1, q0**2-1, q0*q1, q1**2-1])

Or in combination with numpy objects using various arithmetics:

>>> q0**numpy.arange(4)-q1**numpy.arange(3, -1, -1)
polynomial([-q1**3+1, -q1**2+q0, q0**2-q1, q0**3-1])

The constructed polynomials can be evaluated as needed:

>>> poly = 3*q0+2*q1+1
>>> poly(q0=q1, q1=[1, 2, 3])
polynomial([3*q1+3, 3*q1+5, 3*q1+7])

Or manipulated using various numpy functions:

>>> numpy.reshape(q0**numpy.arange(4), (2, 2))
polynomial([[1, q0],
            [q0**2, q0**3]])
>>> numpy.sum(numpoly.monomial(13)[::3])
polynomial(q0**12+q0**9+q0**6+q0**3+1)

Development

Development is done using Poetry manager. Inside the repository directory, install and create a virtual environment with:

poetry install

To run tests:

poetry run pytest numpoly test doc --doctest-modules

To build documentation, run:

cd doc/
make html

The documentation will be generated into the folder doc/.build/html.

Questions and Contributions

Please feel free to file an issue for:

  • bug reporting

  • asking questions related to usage

  • requesting new features

  • wanting to contribute with code

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

numpoly-1.0.0.tar.gz (59.0 kB view hashes)

Uploaded Source

Built Distribution

numpoly-1.0.0-py2.py3-none-any.whl (113.9 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page