从列索引中选择二维数组元素的向量

Question

我有一个二维阵列a：

28  39  52
77  80  66   
7   18  24    
9   97  68

以及列索引B的向量数组：

^{pr2}$

以pythonian的方式，使用基本Python或Numpy，如何从a中选择与B中的列索引不对应的元素？在

我应该得到这个2D数组，它包含A的元素，而不是存储在B中的列索引：

28  52
80  66   
7   18    
97  68

Answer 1

您可以使用广播和行掩码为每行选择数组中未包含的元素：

设置

B = np.array([1, 0, 2, 0])
cols = np.arange(A.shape[1])

---

现在使用广播创建一个掩码，并索引数组。在

^{pr2}$

array([[28, 52],
       [80, 66],
       [ 7, 18],
       [97, 68]])