Select 来自二维 numpy 数组的非连续行和列索引

Question

我有a数组

a = np.arange(5*5).reshape(5,5)

array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

并且想要 select 第一行和第二行的最后两列，以及第三行和第四行的前两列。

结果应该是这样的

array([[3,  4, 10, 11],
       [8,  9, 15, 16]])

如何在一个 go 中做到这一点而无需两次索引和连接？

我尝试使用take

a.take([[0,1,2,3], [3,4,0,1]])
array([[0, 1, 2, 3],
       [3, 4, 0, 1]])

九_

a[np.ix_([0,1,2,3], [3,4,0,1])]
array([[ 3,  4,  0,  1],
       [ 8,  9,  5,  6],
       [13, 14, 10, 11],
       [18, 19, 15, 16]])

和r_

a[np.r_[0:2, 2:4], np.r_[3:5, 0:2]]
array([ 3,  9, 10, 16])

以及 ix_ 和 r_ 的组合

a[np.ix_([0,1,2,3], np.r_[3:4, 0:1])]
array([[ 3,  0],
       [ 8,  5],
       [13, 10],
       [18, 15]])

Answer 1

使用integer 高级索引，你可以做这样的事情

index_rows = np.array([
    [0, 0, 2, 2],
    [1, 1, 3, 3],
])

index_cols = np.array([
    [-2, -1, 0, 1],
    [-2, -1, 0, 1],
])

a[index_rows, index_cols]

你只需 select 直接输入你想要的元素。