/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

1 vote222 views1 answer
Python how often character ins tring
Python
40 votes184 views4 answers
Find todays date in python
Python
4 votes184 views1 answer
Python random true false
Python
0 vote216 views1 answer
Chi square test contingency table python
Python
9 votes201 views2 answers
Get a list of all files python
Python
1 vote218 views1 answer
Panda Series Function
Python
0 vote208 views1 answer
Pyaduio
Python
117 votes202 views6 answers
Python how to get user input
Python
-1 vote192 views1 answer
How to add one 2d list to another 2d list python
Python
3 votes182 views1 answer
If condition
Python