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

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 一次获取NumPy数组中的几个元素的索引，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含953字，纯文字阅读大概需要2分钟。

内容图文

有没有办法一次获取NumPy数组中的几个元素的索引？

例如.

import numpy as np
a = np.array([1, 2, 4])
b = np.array([1, 2, 3, 10, 4])

我想找到a中每个元素的索引,即：[0,1,4].

我发现我使用的解决方案有点冗长：

import numpy as np

a = np.array([1, 2, 4])
b = np.array([1, 2, 3, 10, 4])

c = np.zeros_like(a)
for i, aa in np.ndenumerate(a):
    c[i] = np.where(b==aa)[0]

print('c: {0}'.format(c))

输出：

c: [0 1 4]

解决方法:

您可以使用in1d和nonzero(或者就此而言)：

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

这适用于您的示例数组,但通常返回的索引数组不符合a中值的顺序.这可能是一个问题,具体取决于您下一步要做什么.

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

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

这将返回值中的值的索引.例如：

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]

内容总结

以上是互联网集市为您收集整理的python – 一次获取NumPy数组中的几个元素的索引全部内容，希望文章能够帮你解决python – 一次获取NumPy数组中的几个元素的索引所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端