[Numpy-discussion] Using np.partition to extract n largest/smallest items from an array