/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 vote256 views2 answers
How to map longitude and latitude in python
Python
2 votes278 views3 answers
How to loop over month name in python
Python
2 votes295 views1 answer
Show rows with a null value pandas
Python
0 vote294 views1 answer
Numpy index array all except
Python
0 vote288 views1 answer
Genrate random code in django
Python
0 vote281 views1 answer
Pandas.describe per group
Python
80 votes300 views4 answers
Python append to file
Python
0 vote341 views1 answer
Flask read form data
Python Flask
0 vote291 views1 answer
Find where df series is null and print
Python
-1 vote290 views1 answer
Simple sum in python
Python