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

一次获取NumPy数组中几个元素的索引

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

一次获取NumPy数组中几个元素的索引

您可以使用

in1d
nonzero
(或
where
为此):

>>> np.in1d(b, a).nonzero()[0]array([0, 1, 4])

这对于您的示例数组很好用,但是通常返回的索引数组不遵循中的值顺序

a
。这可能是个问题,具体取决于您下一步要做什么。

在这种情况下,更好的答案是一个@Jaime给出了这里,使用

searchsorted

>>> sorter = np.argsort(b)>>> sorter[np.searchsorted(b, a, sorter=sorter)]array([0, 1, 4])

返回值在中出现的索引

a
。例如:

a = np.array([1, 2, 4])b = np.array([4, 2, 3, 1])>>> sorter = np.argsort(b)>>> sorter[np.searchsorted(b, a, sorter=sorter)]array([3, 1, 0]) # the other method would return [0, 1, 3]


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

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

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