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

【问题标题】：ndarray delete rows by indexes from condition in another arrayndarray 按索引从另一个数组中的条件删除行
【发布时间】：2021-06-02 23:31:07
【问题描述】：

我有两个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]

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

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

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

【问题讨论】：

【解决方案1】：

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

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

输出

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

【讨论】：