/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

15 votes164 views5 answers
Confusion matrix python code
Python
0 vote107 views1 answer
Pytorchvideo cant pickle local object 'kinetics data module.val dataloader.<locals>.<lambda>'
Python
0 vote146 views1 answer
Def form_valid
Python
2 votes174 views3 answers
Selenium find element by link text python
Python
0 vote150 views1 answer
Python how to make boxplots with swarmplot
Python
2 votes172 views3 answers
List to text python
Python
9 votes125 views8 answers
Python get first element from list
Python
0 vote172 views1 answer
Seaborn python for stacked column
Python
0 vote107 views1 answer
Python word encode asci
Python
13 votes201 views4 answers
Update anaconda
Python