npx 0.0.1

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

License

npx is published under the MIT license.