栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

numpy:对于一个数组中的每个元素,在另一个数组中找到索引

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

numpy:对于一个数组中的每个元素,在另一个数组中找到索引

正如Joe
Kington所说,searchsorted()可以非常快速地搜索元素。要处理不在x中的元素,可以用原始y检查搜索结果,并创建一个掩码数组:

import numpy as npx = np.array([3,5,7,1,9,8,6,6])y = np.array([2,1,5,10,100,6])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] != yresult = np.ma.array(yindex, mask=mask)print result

结果是:

[-- 3 1 -- -- 6]


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/639018.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号