我有2个numpy数组。例如
A = [[1,2],[3,4],[5,6]]
B = [[5,6],[1,4],[6,5],[1,2]]
我想获取2个数组中的匹配元素。
即[[1,2],[5,6]]
我在代码中使用的数组非常大。有没有使用for循环和比较每个元素的快速方法?
答案 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])