python - 从numpy数组中查找和删除列

Question

我正在尝试从数组中删除特定列:

data = array([[ 6,  3,  3],
              [ 7,  4, -1]])

target = array([[3]
                [-1]])

我想在 data 内找到该列的索引与 target 相同，以便我可以调用:

np.delete(data, index, axis=1)

看起来像是 all 的某种组合和 where可能是答案，但我就是无法解决。

Answer 1

口罩 (data == target).all(axis=0)将检查沿列的相等性。您不需要对 where 做任何事情或 delete .只需按原样使用掩码:

mask = (data == target).all(axis=0)
data = data[:, ~mask]

事实上，计算不同元素的掩码可能会更快，因为那些是您想要保留的。

mask = (data != target).any(axis=0)
data = data[:, mask]