/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

7 votes281 views1 answer
Pandas insert column in the beginning
Python
60 votes302 views6 answers
Python remove letters from string
Python
2 votes273 views1 answer
Column standardization pandas
Python
3 votes326 views1 answer
Flask upload
Python Flask
0 vote295 views1 answer
Bash check if python package is installed
Python
56 votes313 views5 answers
Python format datetime
Python
0 vote297 views1 answer
Getch backspace pytohn
Python
9 votes285 views2 answers
Django app
Python
4 votes324 views2 answers
Python convert nan to empty string
Python
0 vote236 views3 answers
How to compare the two key from constant value to list of string in python
Python