使用二维数组中的索引从一维数组中获取值

Question

我有一组值

values = np.array([5, 4, 3, 2, 1])

值的索引数组

indices = np.array([[2, 3],
                    [1, 2],
                    [3, 4]])

我想使用索引值作为值的索引并得到结果

np.array([[3, 2],
          [4, 3],
          [2, 1])

我该怎么做？

Answer 1

values[indices]

array([[3, 2],
       [4, 3],
       [2, 1]])