/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

7 votes182 views2 answers
"How to get the remainder of a number when dividing in python"
Python
27 votes206 views6 answers
Create a empty dataframe
Python
2 votes293 views4 answers
Min and max in python
Python
53 votes214 views3 answers
Abc list python
Python
0 vote207 views1 answer
Elongated muskrat
Python
0 vote159 views1 answer
Pyqt5 qpush button pressed
Python
25 votes199 views3 answers
Startapp django
Python
0 vote213 views1 answer
Purge python3.y from every place in my path
Python
0 vote219 views1 answer
Meter to cm in python
Python
0 vote207 views1 answer
Django button
Python