ndarray 通过索引从另一个数组中的条件删除行

问题描述

我有两个ndarrays，其中X的第一个维度的长度与y的大小相同：

X = np.asarray([[1,2,3],[4,5,6],[7,8,9],[3,6,1]])

y = np.asarray([1,3])

我有一个清单：

l = [0,7]

如果来自 X 的同一索引行的值在 y 中，我想删除 l 中的每一行。 因此，在这种情况下，我将：

X = np.asarray([[1,1]])

那是因为来自 y - 的第二个和第三个元素在 l 中。因此，应从 X 中删除第 2 行和第 3 行。 怎么办？

解决方法

一个简单的单行解决方案是使用 delete 和 argwhere

X = np.delete(X,np.argwhere(np.isin(y,l)).flatten(),axis=0)

输出

array([[1,2,3],[3,6,1]])