Getting argsort of numpy array,a.nonzero()