Sort An Array According To Another Array In Python