/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 vote188 views1 answer
Distance of a point from a line python
Python
6 votes190 views6 answers
Ternary operator in python
Python
0 vote186 views1 answer
Randian angle to degrees using numpy
Python
0 vote215 views1 answer
Add url parameter in anchor django template
Python Django
0 vote215 views1 answer
Tuple parameter function python is None
Python
14 votes228 views2 answers
Install python packages in python shell
Python
71 votes193 views6 answers
Shuffle array python
Python
19 votes177 views2 answers
How to add up everything in a list python
Python
0 vote184 views1 answer
Pip install pandas Getting requirements to build wheel
Python
2 votes201 views2 answers
Pandas divide one column by another
Python