/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 vote238 views1 answer
Print lowercase in python
Python
16 votes238 views1 answer
Python reload function in shell
Python
0 vote256 views1 answer
How to add static files in django 3.2.9
Python
1 vote223 views4 answers
Binary search tree in python
Python
1 vote240 views2 answers
Cv2 save video mp4
Python
0 vote240 views1 answer
Python censoring pypi
Python
4 votes238 views2 answers
Make binary tree in python
Python
0 vote222 views1 answer
How remove spaces python input
Python
0 vote219 views1 answer
Python argparse type date
Python
31 votes232 views1 answer
UnicodeDecodeError: 'charmap' codec can't decode byte 0x9d in position 6148: character maps to <undefined>
Python