npx 0.0.2

Some useful extensions for NumPy

npx

NumPy is a large library used everywhere in scientific computing. That's why breaking backwards-compatibility is comes as a significant cost and is almost always avoided, even if the API of some methods is arguably confusing. This package provides drop-in wrappers "fixing" those.

If you have a fix for a NumPy method that can't go upstream for some reason, feel free to PR here.

• npx.dot(a, b)
  

  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).)

• npx.solve(A, b)
  

  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.

• npx.sum_at(a, idx, minlength: int = 0)
  npx.add_at(out, idx, a)
  

  Returns an array with entries of a summed up at indices idx with a minumum 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:

  

  (See also this numpy bug.)

• npx.unique_rows(a, return_inverse=False, return_counts=False)
  

  Returns the unique rows of the array a. The numpy alternative np.unique(a, axis=0) is slow.

License

npx is published under the MIT license.