搜索一个二维阵列中的另一个二维数组在Python [重复]

Question

这个问题已经在这里有一个答案：

• Get intersecting rows across two 2D numpy arrays 7个回答

我有2个numpy的阵列。例如。

A = [[1,2],[3,4],[5,6]]
B = [[5,6],[1,4],[6,5],[1,2]]

我想在2个阵列中的匹配元素。即[[1,2],[5,6]]

我用我的代码数组是非常大的。有没有实现它，而无需使用for循环和比较每个元素的任何快速的方法？

Answer 1

使用列表理解：

A = [[1,2],[3,4],[5,6]]
B = [[5,6],[1,4],[6,5],[1,2]]

print([x for x in A if x in B])
# [[1, 2], [5, 6]]

Answer 2

A = [[1,2],[3,4],[5,6]]
B = [[5,6],[1,4],[6,5],[1,2]]

print([x for x in A if x in B])