How to get the values from a NumPy array using multiple indices,"arr[[0, 1, 1], [1, 0, 2]]"