/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 vote248 views1 answer
R how to merge data frames
Python
0 vote290 views1 answer
Python PyDrive service account credentials
Python
1 vote255 views1 answer
Python multiplication array
Python
3 votes278 views2 answers
Extract one column from dataframe python
Python
0 vote263 views1 answer
Python select from list by condition
Python
1 vote282 views2 answers
Group by dateime pandas
Python
10 votes259 views3 answers
Check python 32 or 64
Python
2 votes329 views2 answers
Python fibonacci while loop
Python
0 vote290 views4 answers
Craeting a method that can take any number of arguments in python
Python
1 vote263 views2 answers
Python print utf-8
Python