考慮 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 npQ = 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 的意思嗎？對我所缺少的有什么見解嗎？
1 回答
滄海一幻覺
TA貢獻1824條經驗 獲得超5個贊
您所描述的所需輸出是：
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]]
添加回答
舉報
0/150
提交
取消