python - 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 - 的第二个和第三个元素在 l 中。因此，应该从 X 中删除第 2 行和第 3 行。怎么做呢？

标签： pythondataframenumpymultidimensional-arraynumpy-ndarray

Answer 1

一个简单的单线解决方案将使用delete和argwhere

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

输出

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