/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 votes166 views1 answer
Average out all rows pandas
Python
2 votes155 views3 answers
Area of a circle in python
Python
0 vote155 views1 answer
Np choose explain
Python
0 vote151 views1 answer
Python import colors
Python
0 vote135 views1 answer
Pandas get size of each group
Python
0 vote132 views1 answer
Recover dict from 0-d numpy array
Python
5 votes133 views2 answers
Random boolean python
Python
0 vote156 views1 answer
Flask session auto logout in 5 mins
Python
10 votes138 views4 answers
Pandas using eval converter excluding nans
Python
0 vote132 views1 answer
AssertRaises property
Python