sorting numpy array according to the sum,"np.take(a, idx, axis=1)"