/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 vote166 views2 answers
Prevent division by zero numpy
Python
4 votes174 views1 answer
Selenium find item by class
Python
1 vote188 views1 answer
Reverse one hot encoding python numpy
Python
0 vote207 views1 answer
Max length of characters in dataframe's columns
Python
0 vote254 views1 answer
Python mann kendall test
Python
0 vote191 views1 answer
ModuleNotFoundError: No module named 'rest_auth'
Python
0 vote188 views1 answer
Blender 2.8 python set active object
Python
0 vote163 views1 answer
Lims
Python
0 vote191 views1 answer
Django.core.exceptions.ImproperlyConfigured: Application labels aren't unique, duplicates: auth
Python
30 votes183 views6 answers
Get python version in code
Python