/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

16 votes312 views1 answer
Which folder python os
Python
0 vote302 views2 answers
Addinput() python
Python
0 vote324 views2 answers
Read_json
Python
1 vote318 views1 answer
How to convert an image to ascii art using python
Python
186 votes401 views6 answers
How to read file in python
Python
0 vote324 views1 answer
Pop element from heap python
Python
0 vote286 views1 answer
Pandas slice of time
Python
21 votes320 views4 answers
Convert radians to degrees python
Python
2 votes335 views1 answer
Python Key–value database
Python
0 vote348 views1 answer
Is power of python recursion
Python