/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

71 votes319 views4 answers
Saving a pandas dataframe as a csv
Python
1 vote287 views1 answer
Create period pandas
Python
0 vote300 views1 answer
How to change the starting number for the loop count in pythin
Python
0 vote325 views1 answer
Tessa thompson
Python
0 vote311 views1 answer
Python notebook breakpoints
Python
0 vote385 views1 answer
Pt_core_news_sm spacy download
Python
1 vote300 views1 answer
Python how to change an element in a multi dimensional list
Python
0 vote338 views1 answer
2nd to last index python
Python
0 vote305 views1 answer
How to read PDF from link in Python]
Python
0 vote367 views1 answer
How to get data from json web api in python
Python