/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

2 votes138 views1 answer
Getting dummies for a column in pandas dataframe
Python
0 vote171 views1 answer
Pass query params django template
Python Django
0 vote122 views1 answer
Operator.itemgetter(1) in python
Python
1 vote133 views1 answer
Python repeating scheduler
Python
6 votes137 views3 answers
Pd df to series
Python
-1 vote144 views1 answer
Let’s add a docstring to the greeting method. How about, “Outputs a message with the name of the person”.
Python
3 votes157 views2 answers
Python - remove scientific notation
Python
1 vote133 views2 answers
How to print without space in python 3
Python
-1 vote205 views3 answers
ValueError: If using all scalar values, you must pass an index
Python
20 votes141 views3 answers
Python download image
Python