是否有一个numpy函数，给出了索引列表，返回了对应于矩阵每一行的元素列表

Question

仅看以下示例：
有2D numpy数组或2D矩阵a：

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

更直观：a是3x3 matrix

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

有一个索引列表：idxes = [2, 1, 1]，它对应于a的每一行，我要返回的是元素列表：
类似于a[idxes] = [a[0][2], a[1][1], a[2][1]] = [3, 5, 8]

array([[1, 2, 3],   2  --> [3,
       [4, 5, 6],   1  -->  5,
       [7, 8, 9]])  1  -->  8]

我知道可以使用for loop来实现它，但是是否有直接的numpy函数或其他更简洁的方法来实现它？