/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

4 votes326 views2 answers
Finding median on python
Python
0 vote327 views1 answer
CharAt in Python
Python
1 vote355 views1 answer
Pyspark Aggregation on multiple columns
Python
-2 votes393 views1 answer
Jwt authentication python flask
Python Flask
3 votes326 views1 answer
Io.UnsupportedOperation: not readable
Python
7 votes298 views1 answer
Python get current file name
Python
2 votes348 views1 answer
Grams in kg
Python
1 vote341 views1 answer
Write csv python pandas stack overflow
Python
26 votes364 views4 answers
Copy text python
Python
0 vote355 views1 answer
Pygame zero how to draw text
Python