/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 vote246 views1 answer
Example of Python Inline comments
Python
0 vote225 views1 answer
Discord embed colors python
Python
0 vote284 views1 answer
Set value through serializer django
Python Django
0 vote251 views1 answer
<function chr(i, /)> error in python
Python
0 vote254 views1 answer
Merge 2 or more notebooks into one
Python
1 vote250 views2 answers
Python dictonary set default
Python
0 vote280 views1 answer
Line plot python only years datetime index
Python
2 votes244 views1 answer
Flask if statement
Python
1 vote262 views1 answer
How to convert days into seconds in python using time.time()
Python
0 vote257 views1 answer
How to access a file from root folder in python project
Python