Search a 2D array in another 2D array in Python

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!! https://stackoverflow.com/questions/8317022/get-intersecting-rows-across-two-2d-numpy-arrays — Leo, Feb 10 '19 at 14:54

score 2 · Answer 1 · answered Feb 07 '19 at 16:21

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 07 '19 at 16:21

Austin

score -1 · Answer 2 · edited Feb 07 '19 at 18:11

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

edited Feb 07 '19 at 18:11

d_kennetz

answered Feb 07 '19 at 16:22

Giridhar

2 Answers2