/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 vote314 views1 answer
Renpy quickstart
Python
0 vote303 views1 answer
Randomly splits this DataFrame with the provided weights
Python
5 votes287 views1 answer
Python check file extension
Python
21 votes314 views2 answers
List comprehension if-else
Python
0 vote333 views1 answer
Day 19: interfaces hackerrank solution in python
Python
2 votes322 views2 answers
Roots of quadratic equation in python
Python
0 vote298 views1 answer
Python import list from py file
Python
0 vote340 views2 answers
Pandas split tuple column
Python
10 votes307 views1 answer
How to get number after decimal point
Python
1 vote318 views2 answers
Phi
Python