Python Sort Array According To Another Array