/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 vote186 views1 answer
How to draw squircle python
Python
4 votes161 views1 answer
Python format only 1 decimal place
Python
0 vote173 views1 answer
Grapper
Python
0 vote164 views1 answer
How to send variable to python using xlwings
Python
7 votes195 views5 answers
Django messages
Python Django
0 vote160 views1 answer
Python while variable is not
Python
0 vote211 views1 answer
Quill get content flask
Python Flask
6 votes180 views2 answers
Kill port django
Python Django
0 vote163 views2 answers
Check status code urllib
Python
0 vote168 views1 answer
How to change canvas background color in python tkinter
Python