关于在 np.where 语句中使用 np.logical_and 的问题

Question

考虑 4 列数组 Q 的一个小样本。我的目标是消除 Q 的行，其中BOTH columns 0 and 2 are equal, AND columns 1 and 3 are equal。所以，[1, 2, 1, 2]需要去，也一样[3, 4, 3, 4]。我认为下面的代码会起作用。

import numpy as np

Q = np.array([[1, 2, 1, 2],

              [1, 2, 1, 4],

              [2, 4, 3, 4],

              [3, 4, 3, 4],

              [4, 7, 2, 5]])

output = Q[np.where(np.logical_and(Q[:,0] != Q[:, 2], Q[:,1] != Q[:, 3]))]

print(output)

[[4 7 2 5]]

此代码消除了所需的两行，但它也删除了[1, 2, 1, 4]和[2, 4, 3, 4]。

换句话说，它正在做：

“如果 V[:,0] != V[:, 2] 则删除记录，如果 V[:,1] != V[:, 3] 也删除记录”。

我希望它：“如果 V[:,0] != V[:, 2] 和 V[:,1] != V[:, 3] 都删除记录”。

这不是 logical_and 的意思吗？

对我所缺少的有什么见解吗？

Answer 1

您所描述的所需输出是：output = Q[~((Q[:,0]==Q[:,2]) * (Q[:,1]==Q[:,3]))]或者同样：output = Q[~(np.logical_and(Q[:,0]==Q[:,2], Q[:,1]==Q[:,3]))]或者同样：output = Q[np.logical_or(Q[:,0]!=Q[:,2], Q[:,1]!=Q[:,3])]或者同样：output = Q[(Q[:,0]!=Q[:,2]) + (Q[:,1]!=Q[:,3])]但我更喜欢这种方式（这可能更快）：output = Q[(Q[:,0:2]!=Q[:,2:]).any(1)]请注意，您不需要np.where。布尔掩码适用于数组的行Q。输出：[[1 2 1 4]  [2 4 3 4]  [4 7 2 5]]