使用2d数组Python索引2d数组

Question

我有2d Numpy数组arr1，其形状为（2,5），以及arr2，其形状为（2,2）。 我希望arr2的第一行索引arr1的第一行，类似 arr2的第二行将索引到arr1，因此最终形状为（2,2）。

如何有效地做到这一点，特别是没有循环？ TIA。 例如：

  

arr1 Out[19]:
array([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]) arr2 Out[20]:
array([[0, 3], [1, 2]]) required Out[21]: array([[0, 3], [6, 7]])