/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

92 votes135 views3 answers
Renaming column in dataframe pandas
Python
1 vote152 views1 answer
Tkinter boilerplate
Python
4 votes145 views1 answer
How to make a text input box python pygame
Python
1 vote145 views1 answer
How to run commands in repl.ot
Python
-3 votes159 views3 answers
Chatterbot python
Python
1 vote141 views1 answer
How to select a file in python
Python
0 vote186 views1 answer
Smtplib send caleneder email
Python
0 vote131 views1 answer
Open gui window python
Python
0 vote139 views1 answer
Newsapi in python
Python
0 vote142 views1 answer
Dt.weekday_name
Python