Search a 2D array in another 2D array in Python [duplicate]

Question

I have 2 numpy arrays. Eg.

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

I want to get the matching elements in the 2 arrays. i.e [[1,2],[5,6]]

The arrays that I am using in my code are very large. Is there any fast way of implementing it, without using for loop and comparing each elements?

This fixed my problem. Thanks!! stackoverflow.com/questions/8317022/… — Leo
– Leo, Commented Feb 10, 2019 at 14:54

Austin · Accepted Answer · 2019-02-07 16:21:19Z

Use a list-comprehension:

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]]

d_kennetz · Accepted Answer · 2019-02-07 18:11:19Z

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

5,3975 gold badges24 silver badges49 bronze badges

answered Feb 7, 2019 at 16:22

466 bronze badges

2 Answers 2