Python Sort One Array Based On Another