Numpy int array: Find indices of multiple target ints

后端 未结 3 475
孤独总比滥情好
孤独总比滥情好 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:26

    Can you avoid numpy all together? List concatenation should be much faster than relying on numpy's methods. This will still work even if values needs to be a numpy array.

    result = []
    for sv in searchvals:
        result += [i for i in range(len(values)) if values[i] == sv]
    

提交回复
热议问题