python numpy keep a list of indices of a sorted 2d array,i = a.argsort(axis=None)[::-1]