获取numpy数组中元素的索引

可以使用numpy库中的函数来实现。具体的函数是numpy的where函数。

where函数可以根据给定的条件返回满足条件的元素的索引。它的语法如下：

numpy.where(condition[, x, y])

参数说明：

• condition：条件表达式，可以是一个布尔数组或者布尔表达式。
• x：可选参数，满足条件的元素将被替换为x中对应位置的元素。
• y：可选参数，不满足条件的元素将被替换为y中对应位置的元素。

返回值是一个元组，包含满足条件的元素的索引。

下面是一个示例：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
indices = np.where(arr == 3)

print(indices)

输出结果为：

(array([2]),)

这表示元素3在数组中的索引为2。

对于多维数组，where函数也可以使用。它会返回满足条件的元素的索引的元组，其中每个元素都是一个数组，分别表示对应维度上的索引。

除了where函数，还可以使用argwhere函数来获取满足条件的元素的索引。argwhere函数的用法与where函数类似，但返回的是一个二维数组，每一行表示一个满足条件的元素的索引。

希望以上信息对您有所帮助。如果您需要了解更多关于numpy的知识，可以参考腾讯云的相关产品和文档：

• 腾讯云产品介绍：云服务器
• 腾讯云产品介绍：云数据库
• 腾讯云产品介绍：云原生应用引擎
• 腾讯云产品介绍：人工智能
• 腾讯云产品介绍：物联网
• 腾讯云产品介绍：移动开发
• 腾讯云产品介绍：对象存储
• 腾讯云产品介绍：区块链
• 腾讯云产品介绍：元宇宙