npx
NumPy is a large library used everywhere in scientific computing. That's why breaking backwards-compatibility comes at a significant cost and is almost always avoided, even if the API of some methods is arguably lacking. This package provides drop-in wrappers "fixing" those.
scipyx does the same for SciPy.
If you have a fix for a NumPy method that can't go upstream for some reason, feel free to PR here.
dot
import npx
import numpy as np
a = np.random.rand(3, 4, 5)
b = np.random.rand(5, 2, 2)
out = npx.dot(a, b)
# out.shape == (3, 4, 2, 2)
Forms the dot product between the last axis of a and the first axis of b.
(Not the second-last axis of b as numpy.dot(a, b).)
np.solve
import npx
import numpy as np
A = np.random.rand(3, 3)
b = np.random.rand(3, 10, 4)
out = npx.solve(A, b)
# out.shape == (3, 10, 4)
Solves a linear equation system with a matrix of shape (n, n) and an array of shape
(n, ...). The output has the same shape as the second argument.
sum_at/add_at
npx.sum_at(a, idx, minlength=0)
npx.add_at(out, idx, a)
Returns an array with entries of a summed up at indices idx with a minimum length of
minlength. idx can have any shape as long as it's matching a. The output shape is
(minlength,...).
The numpy equivalent numpy.add.at is much
slower:
Relevant issue reports:
unique
import npx
a = [0.1, 0.15, 0.7]
a_unique = npx.unique(a, tol=2.0e-1)
assert all(a_unique == [0.1, 0.7])
npx's unique() works just like NumPy's, except that it provides a parameter
tol (default 0.0) which allows the user to set a tolerance. The real line
is essentially partitioned into bins of size tol and at most one
representative of each bin is returned.
unique_rows
import npx
import numpy as np
a = np.random.randint(0, 5, size=(100, 2))
npx.unique_rows(a, return_inverse=False, return_counts=False)
Returns the unique rows of the integer array a. The numpy alternative np.unique(a, axis=0) is slow.
Relevant issue reports:
isin_rows
import npx
import numpy as np
a = [[0, 1], [0, 2]]
b = np.random.randint(0, 5, size=(100, 2))
npx.isin_rows(a, b)
Returns a boolean array of length len(a) specifying if the rows a[k] appear in b.
Similar to NumPy's own np.isin which only works for scalars.
mean
import npx
a = [1.0, 2.0, 5.0]
npx.mean(a, p=3)
Returns the generalized mean of a
given list. Handles the cases +-np.inf (max/min) and0 (geometric mean) correctly.
Also does well for large p.
Relevant NumPy issues:
License
This software is published under the BSD-3-Clause license.
1 maintainer
Metadata
1 maintainer
Release files for npx 0.1.8
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| npx-0.1.8.tar.gz | 10.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| npx-0.1.8-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 19.2 kB
Release files / npx-0.1.8.tar.gz
| Download URL | npx-0.1.8.tar.gz |
|---|---|
| Size | 10.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
efcedd9f8090864c1ad154307281302e931dfab60b495fada78bb505ac884eb9
|
|
BLAKE2b-256 checksum How to use checksums |
217adfb51fe0a80e08ff4ecdb6b4f079e43a778eef6e79568a416077c18858e6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.11.8 {"installer":{"name":"uv","version":"0.11.8","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
Release files / npx-0.1.8-py3-none-any.whl
| Download URL | npx-0.1.8-py3-none-any.whl |
|---|---|
| Size | 8.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
9263f7e54da5c50f2fe1757b64d0dc805e65325b3b3af7fbca93a89b06231509
|
|
BLAKE2b-256 checksum How to use checksums |
9eb44987134ebb73f127775ef6a39af55f5ba1edb7fc55cadcca58640728668e
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.11.8 {"installer":{"name":"uv","version":"0.11.8","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
