Skip to content
Advertisement

Find whether 1D sub-array is in 2D array

I have a data-array like:

JavaScript

It has about 4000 pairs in it, but typically fewer on the order of a few hundred.

I need to find whether a two-valued array already exists in this large data-array, eg. does [1,1] exist in the array?

So the function I need should act something like this:

JavaScript

I couldn’t get the numpy functions isin() or in1d() to do this for me. The one function I did find works, for lists, is:

JavaScript

but when the arg is not in the all array, I have to try/catch a ValueError and then return False – acceptable for now, but not ideal.

Advertisement

Answer

You can use simple array lookup like this:

JavaScript

or

JavaScript
User contributions licensed under: CC BY-SA
5 People found this is helpful
Advertisement