在Python中的另一个2D数组中搜索2D数组

Question

我有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])