Python 如何基于另一个numpy数组的值从numpy数组中删除行?
我有一个Python 如何基于另一个numpy数组的值从numpy数组中删除行?,python,python-3.x,numpy,Python,Python 3.x,Numpy,我有一个标签的向量,如图所示: labels2 = np.array([0,-1,1,2,-1]) 我想从X\u train中删除对应于标签中值为-1的行的行 X_train = np.arange(1,16).reshape(5,3) 输出应该是这样的 X_train_trimmed = np.array([[ 1 2 3] [ 7 8 9] [10 11 12]])
标签的向量
,如图所示:
labels2 = np.array([0,-1,1,2,-1])
我想从X\u train
中删除对应于标签中值为-1的行的行
X_train = np.arange(1,16).reshape(5,3)
输出应该是这样的
X_train_trimmed = np.array([[ 1 2 3]
[ 7 8 9]
[10 11 12]])
我应该怎么做呢?只需按如下所示进行简单的索引,然后重新分配新结果:
In [54]: X_train = X_train[labels2 != -1]
In [55]: X_train
Out[55]:
array([[ 1, 2, 3],
[ 7, 8, 9],
[10, 11, 12]])
只需按如下所示执行简单的索引,然后重新分配新结果:
In [54]: X_train = X_train[labels2 != -1]
In [55]: X_train
Out[55]:
array([[ 1, 2, 3],
[ 7, 8, 9],
[10, 11, 12]])
这比我想象的要简单得多!非常感谢。这比我想象的要简单得多!非常感谢。