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