/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 vote186 views1 answer
Voting classifier grid search
Python
0 vote388 views1 answer
Get sum of column before a date python
Python
0 vote219 views1 answer
Def get_label(Dir): for nextdir in os.listdir(Dir): if not nextdir.startswith('.'): if nextdir in ['NORMAL']: label = 0 elif nextdir in ['PNEUMONIA']: label = 1 else: label = 2 return nextdir, label
Python
2 votes180 views2 answers
Pandas shift columns down until value
Python
0 vote175 views1 answer
How to split text in a csv file in dataset python
Python
0 vote171 views1 answer
2checkout python
Python
2 votes158 views2 answers
Hwo to separate datetime column into date and time pandas
Python
1 vote173 views1 answer
Get all keys and values from dictionary python
Python
0 vote180 views1 answer
How to make software in python
Python
4 votes169 views2 answers
Code server
Python