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

时间:2019-02-07 16:17:01

标签: python numpy

我有2个numpy数组。例如

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

我想获取2个数组中的匹配元素。 即[[1,2],[5,6]]

我在代码中使用的数组非常大。有没有使用for循环和比较每个元素的快速方法?

2 个答案:

答案 0 :(得分: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])
# [[1, 2], [5, 6]]

答案 1 :(得分:-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])