/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

22 votes110 views4 answers
Python is float
Python
1 vote106 views1 answer
How to convert tuple into list in python
Python
1 vote91 views1 answer
How to get input from user in pyqt5
Python
0 vote108 views1 answer
Flask recive list
Python
13 votes104 views5 answers
How to make a timer using python
Python
0 vote120 views1 answer
Perform a right outer join of self and other.
Python
21 votes92 views2 answers
How do a plot on matplotlib python
Python
106 votes102 views6 answers
Generate random int python
Python
0 vote95 views1 answer
How to estimate process timing python
Python
0 vote104 views1 answer
Pop function second argument in python
Python