python - 在二维 Numpy 数组中切片非连续索引

Question

我有一个矩阵，即 2D numpy 数组，希望能够切出它的非连续部分。例如使用以下矩阵

[[11 12 13 14 15]
 [21 22 23 24 25]
 [31 32 33 34 35]
 [41 42 43 44 45]
 [51 52 53 54 55]]

我希望能够提取，说

[[11 12 14 15]
 [21 22 24 25]
 [51 52 54 55]]

有没有办法做到这一点？
我可以轻松提取连续切片，例如 matrix[0:2,0:3] 将返回

[[11 12 13]
 [21 22 23]

但不确定如何提取非连续。我阅读了关于使用 np.r_[] 的信息，它仅在一维上使用时有效，但在两个维上均无效。

该解决方案需要具有可扩展性，并且必须用于具有许多非连续索引的大型矩阵(我认为将作为列表传递)。

Answer 1

您可以使用链式索引:

arr = np.array([[11, 12, 13, 14, 15],
       [21, 22, 23, 24, 25],
       [31, 32, 33, 34, 35],
       [41, 42, 43, 44, 45],
       [51, 52, 53, 54, 55]])

In [28]: arr[[0,1,4]][:, [0,1,3,4]]                                                                       
Out[28]: 
array([[11, 12, 14, 15],
       [21, 22, 24, 25],
       [51, 52, 54, 55]])