/Python/Numpy find index of matching values

Numpy find index of matching values

import numpy as np
# init arrays
a = np.array([1,2,3,2,3,4,3,4,5,6])
b = np.array([7,2,10,2,7,4,9,4,9,8])

#using enumerate, list comprehension and set
print([key for key, val in enumerate(a) if val in set(b)])
# output
# [1, 3, 5, 7]

-2

Similar codes

0 vote276 views1 answer
Index from multiindex pandas
Python
16 votes271 views5 answers
How to create window in tkinter
Python
2 votes286 views2 answers
Python count total no of word in a text
Python
11 votes284 views3 answers
Pandas conditional replace values in a series
Python
1 vote329 views1 answer
Django models.py convert DateTimeField to DateField
Python Django
0 vote284 views2 answers
How to fix def multiply(a ,b): a*b
Python
0 vote279 views1 answer
Setting price variable in 3 categories python
Python
0 vote282 views1 answer
Studentized residuals python
Python
2 votes284 views2 answers
How to add time with time delta in python
Python
-3 votes294 views6 answers
Python 3.3 release date
Python