Python Sort 2d Numpy Array To 1d