Numpy int array: Find indices of multiple target ints

后端 未结 3 510
孤独总比滥情好
孤独总比滥情好 2021-01-18 07:46

I have a large numpy array (dtype=int) and a set of numbers which I\'d like to find in that array, e.g.,

import numpy as np
values = np.array(         


        
3条回答
  •  孤独总比滥情好
    2021-01-18 08:23

    I would say using np.in1d would be the intuitive solution to solve such a case. Having said that, based on this solution here's an alternative with np.searchsorted -

    sidx = np.argsort(searchvals)
    left_idx = np.searchsorted(searchvals,values,sorter=sidx,side='left')
    right_idx = np.searchsorted(searchvals,values,sorter=sidx,side='right')
    out = np.where(left_idx != right_idx)[0]
    

提交回复
热议问题