Python-数组之间的重复对比
如果我有两个数组：
A=[1,2,3,4,5,6,7]
B=[2,4,7]
我想获得一个数组C，它包含在A中也可以找到的B的值的索引。
C=[1,3,6]
我对Python还是很陌生，对于无法在不使用结合numpy.where（）的循环的情况下找到如此简单的任务的优雅解决方案感到沮丧。
提前致谢！！
问题来源：stackoverflow
展开
收起
1
条回答
写回答
写回答
-
这是线性时间解决方案：要有效测试元素是否在B中，请先将其转换为集合。
B_set = set(B) C = [i for i, x in enumerate(A) if x in B_set]对于大型输入，这比在循环中使用.index更好，因为这需要在O（mn）时间内重复搜索列表，其中m和n是A和B的大小。相比之下，上述解决方案需要O（m + n）时间转换为一个集合，然后构建结果列表。
回答来源：stackoverflow
2020-03-24 19:29:23赞同 展开评论 打赏
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
相关问答
问答排行榜
最热
最新
推荐问答
相关文章
相关电子书
更多
