numpy的searchsorted( )用于在两个数组中查找相同元素

792866506

import numpy as np
def get_index_searchsorted(x,y):
    index=np.argsort(x)
    sorted_x=x[index]
    sorted_index=np.searchsorted(sorted_x,y)
    yindex=np.take(index,sorted_index,mode='clip')
    mask= x[yindex]!=y
    yindex[mask]=-1
    return yindex
x=np.array([ 3,  5,  7,  1,  9,  8,  6, 10])
y=np.array([  2,   1,   5,  10, 100,   6])
get_index_searchsorted(x,y)

有x,y两个数组，找到y中每个元素在x中的下标，若不存在则下标设置为-1。
算法有点复杂，没太看懂。

lumber2388779

np.argsort(x)是将x从小到大排序的索引值组合新的列表返回
sorted_x=x[index]就是将x从小到大排序的列表赋值给sorted_x
np.searchsorted(sorted_x,y)搜索y中每个元素在sorted_x的索引值组合成新的列表赋值给sorted_index
np.take(index,sorted_index,mode='clip')取出index中索引值位sorted_index的值赋值给yindex
mask= x[yindex]!=y 拿y与x中索引出的值进行比较形成布尔类型列表赋值给mask
yindex[mask]=-1将mask中值为false的赋值为-1

其实你每一句都将结果打印出来 你就可以理解了

SixPy

在两个数组中查找相同元素
直接这样不是更简单？

>>> set(x)&set(y)
{1, 10, 5, 6}