Python 求两个列表公共元素及其在各自中的索引

遇到一个小小的需求:现在有两个numpy.array,它俩的长度不一致,但是有公共元素,想实现求它们的公共元素在它们各自中的索引。

基本思想是:先求出两个array的公共元素,再遍历元素,得到其在各自中的索引。

下面是实现代码

def get_uionInd(a,b):
    m=list(set(a) & set(b))
    alist=a.tolist()
    blist=b.tolist()
    p=[]
    q=[]
    for i in m:
        p.append(alist.index(i))
        q.append(blist.index(i))
    return np.array(m),p,q

这个方法有点慢,主要有一个for循环,目前没有更好的解决方案。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

流浪猪头拯救地球

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值