/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

1 vote317 views2 answers
String array to float array python
Python
2 votes386 views1 answer
How to make a python program to count from 1 to 100
Python
1 vote329 views1 answer
Time it in jupyter notebook
Python
13 votes329 views2 answers
Map function using lambda in python
Python
146 votes398 views6 answers
Os file exists
Python
-1 vote322 views1 answer
Matrix python math
Python
10 votes324 views4 answers
Pandas
Python
0 vote363 views1 answer
Find a file in python
Python
0 vote312 views1 answer
If short for python
Python
-1 vote334 views3 answers
Count string python
Python