/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 vote129 views1 answer
Derivative of multivariable function pytorch
Python
5 votes190 views3 answers
How to run flask in port 80
Python Flask
29 votes129 views3 answers
System calls in python
Python
4 votes129 views5 answers
'datetime' has no attribute 'now'
Python
2 votes135 views1 answer
Drop a range of rows pandas
Python
0 vote128 views1 answer
While importing we detected an older version of numpy in
Python
3 votes132 views1 answer
Python nmap
Python
1 vote133 views4 answers
Driver.find_element_by_xpath
Python
6 votes126 views2 answers
Tkinter label border
Python
22 votes128 views2 answers
How to plotting points on matplotlib
Python