Numpy Sort Array Based On Another