/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 vote210 views1 answer
Sum() function in Python
Python
1 vote199 views1 answer
Scikit learn ridge classifier
Python
0 vote213 views1 answer
Matrix diagonal sum leetcode in java
Python
37 votes230 views1 answer
Code how pandas save csv file
Python
0 vote200 views1 answer
Check how many letters in a string python
Python
0 vote212 views1 answer
How to save a python object in a file
Python
0 vote254 views1 answer
Explorer gives new window
Python
0 vote219 views1 answer
Pretty print list python
Python
5 votes208 views2 answers
Selenium scroll to element python
Python
0 vote226 views1 answer
What is certifi module in python
Python