Numpy Sort Two Arrays Based On One