Skip to main content

This is a user-friendly wrapper to numpy arrays, with batteries included and numba-enhanced performance.

Project description

Vectorized2D

Tests with conda Tests with pip

codecov

This is a user-friendly wrapper to numpy arrays, for dealing with numerical problems in a vectorized fashion - in the 2D world.

Provided objects include:

  1. Array2D - a user-friendly interface to numpy arrays of shape=Nx2.
  2. Vector2D - a user-friendly wrapper for arrays of 2D vectors that represent physical quantities.
  3. Point2D - a user-friendly wrapper to arrays of 2D points that represent spatial locations in a cartesian coordinate system.
  4. Coordinate - a user-friendly wrapper for arrays of 2D points that represent 2D spatial (geographical) coordinates (longitude and latitude) in radians.

Installation

The easiest way to install vectorized2d and get updates is via the Python Package Index (PyPI):

$ pip install vectorized2d

Performance

Vectorized2d uses Numba to gain enhanced performance compared to vanilla numpy.

For example, (per-row) norm calculation:

  >>> import numpy as np
  >>> from vectorized2d import Array2D

  >>> a_np = np.random.random(size=(1000, 2))
  >>> a_2d = a_np.view(Array2D)

  >>> %timeit np.linalg.norm(a_np, axis=1)
  23.1 µs ± 1.25 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

  >>> %timeit np.sqrt(np.einsum('ij,ij->i', a_np, a_np))
  10.5 µs ± 40.2 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

  >>> %timeit a_2d.norm
  2.63 µs ± 67.9 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

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

vectorized2d-0.0.7.tar.gz (8.8 kB view hashes)

Uploaded Source

Built Distribution

vectorized2d-0.0.7-py3-none-any.whl (10.6 kB view hashes)

Uploaded 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