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

2

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

answered Feb 7, 2019 at 16:21

Austin

26.1k4 gold badges28 silver badges52 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

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

d_kennetz

5,3975 gold badges24 silver badges49 bronze badges

answered Feb 7, 2019 at 16:22

Giridhar

466 bronze badges