numpy怎么获取元素索引

avatar
作者
筋斗云
阅读量:0

numpy中,可以使用numpy.where()函数来获取满足条件的元素的索引。这个函数将返回一个包含满足条件的元素索引的元组。

例如,我们有一个numpy数组arr

import numpy as np  arr = np.array([1, 2, 3, 4, 5]) 

我们想获取所有大于2的元素的索引,可以使用numpy.where()函数:

indexes = np.where(arr > 2) print(indexes) 

输出:

(array([2, 3, 4]),) 

这表示满足条件的元素的索引是2、3和4。如果我们想直接获取索引值而不是一个元组,可以使用numpy.where()的第一个元素:

indexes = np.where(arr > 2)[0] print(indexes) 

输出:

[2 3 4] 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!