如何根据另一个numpy数组中的值从numpy数组中删除行？

Question

我有labels的向量，如下所示：

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

我想从X_train中删除与labels中值为-1的行相对应的行。

X_train = np.arange(1,16).reshape(5,3)

输出应该如下所示。

X_train_trimmed = np.array([[ 1  2  3]
                            [ 7  8  9]
                            [10 11 12]])

我应该怎么做？

Answer 1

只需执行如下的简单索引，然后重新分配新结果：

Candidate: 1, 6, 10