/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

3 votes138 views1 answer
Email authentication python
Python
0 vote132 views1 answer
Take substring of every element in dataframe
Python
5 votes114 views4 answers
Python debugger
Python
0 vote109 views1 answer
No module named 'selenium.webdriver.common.action_chain'
Python
-1 vote149 views1 answer
Flask set mime type
Python Flask
27 votes105 views6 answers
Remove column from dataframe
Python
0 vote117 views1 answer
Make tkinter text editing disabled
Python
1 vote112 views1 answer
Matplotlib change font
Python
0 vote108 views1 answer
IntersectAll dynamo revit
Python
0 vote116 views2 answers
Django edit object foreign key id
Python