How to get indicies of numpy array that equal any values of another array? [duplicate]

Question

I have a numpy array A, of say length 100, and another array B, of length 4.

I want to get a mask of also length 100, and this mask will have values of TRUE, when the element of A equals ANY of the elements of B, and FALSE otherwise. How can I do that in an efficient way?

EduGord · Accepted Answer · 2017-03-05 22:43:36Z

0

Try:

a = np.random.rand(100)
b = np.random.rand(4)
print([(element in b) for element in a])
print('\n')
print([(element in a) for element in b])

I think that's what you want

answered Mar 5, 2017 at 22:43

EduGord

1572 silver badges13 bronze badges

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

1 Comment

Spacey Over a year ago

Thanks, although I would rather do it without a for-loop. Apparently this works `np.in1d'

Collectives™ on Stack Overflow

How to get indicies of numpy array that equal any values of another array? [duplicate]

1 Answer 1

1 Comment

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Linked

Related