/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 vote119 views1 answer
Python get unique pairs from two lists
Python
0 vote114 views1 answer
Printing hollow triangle in python
Python
110 votes111 views6 answers
Read text file in python
Python
0 vote188 views1 answer
How to install flask module in vscode
Python Flask
0 vote123 views1 answer
Racine carré python
Python
0 vote120 views1 answer
Get first letter of each word in string python
Python
0 vote116 views1 answer
Cv2.cv2' has no attribute 'face_lbphfacerecognizer'
Python
0 vote132 views1 answer
Python argparse only allow certain values
Python
9 votes117 views1 answer
How to save image opencv
Python
6 votes131 views2 answers
Flask get ip address of request
Python