/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 vote338 views1 answer
(Word or Phrase to Phone-Number Generator) python
Python
0 vote351 views1 answer
Python var_dump
Python
1 vote349 views1 answer
Append a dataframe to an empty dataframe
Python
0 vote357 views1 answer
How to save the command result with ! in python
Python
0 vote386 views1 answer
Flask decorator causes views to be named the same thing
Python
1 vote379 views1 answer
Remove all pyc files
Python
3 votes356 views2 answers
Uses specific version python venv
Python
1 vote337 views1 answer
Def __init__ python not overwrite parrent class
Python
1 vote338 views1 answer
Assert len(lex) < self.bucket_specs[-1][1]
Python
2 votes314 views1 answer
Generate 12 random numbers python
Python