I am attempting to partition a matrix using np.argpartition.
This works fine for numpy matrices but not for scipy compressed csr matrix.
I am attempting to partition it in axis 1 but since a scipy csr matrix becomes a vector I am facing issues. Is there an analogous function to np.argpartition in scipy?