/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

11 votes257 views1 answer
Selecting items in a column of a dataframe
Python
0 vote243 views1 answer
How to get one record in django
Python
4 votes298 views1 answer
How to make a text input box python pygame
Python
2 votes212 views2 answers
Int' object is not subscriptable in python
Python
0 vote267 views1 answer
How to add import pydictionary in python
Python
6 votes268 views1 answer
Write a Python program to append text to a file and display the text.
Python
2 votes248 views5 answers
Torch.stack
Python
0 vote246 views1 answer
Switch between frames in tkinter
Python
3 votes245 views1 answer
How to exit command prompt
Python
0 vote253 views1 answer
Gspread_pandas pypi
Python