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

Question

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

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

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

我有一个列表：

l = [0, 2, 7]

我想删除 X 中的每一行，如果 y 中相同索引的行的值在 l 中。所以在那种情况下，我将有：

X = np.asarray([[1, 2, 3],
                [3, 6, 1]])

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

Answer 1

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

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

输出

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

ndarray delete rows by indexes from condition in another array