如何使用numpy查找数组中元素的索引? [复制]

Posted

技术标签:

【中文标题】如何使用numpy查找数组中元素的索引? [复制]【英文标题】:How to find an index of an element in an array using numpy? [duplicate] 【发布时间】:2015-10-16 10:56:28 【问题描述】:

如果让我们设置a=array([1,4,5,57,45,34]) 那么我们如何获得一个元素的索引呢?说 5 等等。

我知道如何使用 list.index(ele) 获取列表中的索引。 但是数组呢?有没有类似的功能,.index 绝对不能像我尝试过的那样工作?还是我们自己开发功能?

还有其他类似的问题使用perl,C等,但我在这个社区中没有找到使用python的问题。

【问题讨论】:

【参考方案1】:

如果你想要元素 5 的所有索引,你可以使用 numpy.where ,示例 -

In [1]: from numpy import array

In [2]: a=array([1,4,5,57,45,34])

In [3]: from numpy import where

In [4]: where(a==5)
Out[4]: (array([2], dtype=int64),)

【讨论】:

以上是关于如何使用numpy查找数组中元素的索引? [复制]的主要内容,如果未能解决你的问题,请参考以下文章

在 NumPy 数组中查找等于零的元素的索引

科学计算基础软件包NumPy入门讲座:操作数组

使用花哨的索引从 Numpy 数组中查找和删除全零列

科学计算基础软件包NumPy入门讲座:操作数组

如何在 numpy / scipy 中获取特定百分位数的索引?

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