Python Numpy - if array is in array

Question

I want to check with most efficiency way (the fastest way), if some array (or list) is in numpy array. But when I do this:

import numpy

a = numpy.array(
    [
        [[1, 2]],
        [[3, 4]]
    ])

print([[3, 5]] in a)

It only compares the first value and returns True

Somebody knows, how can I solve it? Thank you.

Does this answer your question? Iterating over Numpy matrix rows to apply a function each? — anurag
– anurag, Commented Jan 27, 2021 at 15:26
for your original data, is only the matrix a large or the query matrix is also large? — anurag
– anurag, Commented Feb 1, 2021 at 13:23

Leonardo Viotti · Accepted Answer · 2021-01-27 15:32:42Z

1

Your question seems to be a duplicate of: How to match pairs of values contained in two numpy arrays

In any case, something like the first answer should do it if I understand correctly:

import numpy

a = numpy.array(
    [
        [[1, 2]],
        [[3, 4]]
    ])

b = numpy.array([[3,5]])

print((b[:,None] == a).all(2).any(1))

Which outputs:

array([False,  True])

answered Jan 27, 2021 at 15:32

Leonardo Viotti

5061 gold badge7 silver badges18 bronze badges

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

1 Comment

Foreen Over a year ago

Thank you! This works, it is faster than solution from joostblack, but it still takes a long time on my big data series. Is there any other faster solution? Thank you.

joostblack · Accepted Answer · 2021-01-27 15:29:52Z

1

You could just add tolist() in your last line:

print([[3, 5]] in a.tolist())

gives

False

answered Jan 27, 2021 at 15:29

joostblack

2,5457 silver badges14 bronze badges

2 Comments

joostblack Over a year ago

I do not know what your application is but I think you can drop the double brackets and instead use single brackets.

Foreen Over a year ago

This works! Thank you, but it is really slow. I have big data series and with them it takes more than 1 minute. Is there any faster solution?

Collectives™ on Stack Overflow

Python Numpy - if array is in array

2 Answers 2

1 Comment

2 Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

2 Comments

Linked

Related