在numpy的ndarray類型中，似乎沒有直接返回特定索引的方法，我只找到了where函數(shù)，但是where函數(shù)對于尋找某個(gè)特定值對應(yīng)的索引很有用，對于返回一定區(qū)間內(nèi)值的索引不是很有效，至少我沒有弄明白應(yīng)該如何操作尷尬。下面先說一下where函數(shù)的用法吧。

（1）where函數(shù)的使用場景：

例如現(xiàn)在我生成了一個(gè)數(shù)組：

import numpy as np 
arr=np.array([1,1,1,134,45,3,46,45,65,3,23424,234,12,12,3,546,1,2]) 

現(xiàn)在arr是一個(gè)含有18個(gè)元素的ndarray類型的數(shù)組，后面就用數(shù)組來稱呼它吧。假如我想返回?cái)?shù)組中所有元素值為3所對應(yīng)的索引位置，那么通過where函數(shù)可以很輕松地完成。

print np.where(arr==3) 

它會返回一個(gè)包含所有值為3對應(yīng)的索引位置的元組，如下圖：

可以看到在索引為5、9、14的地方所對應(yīng)的元素值為3。通過這樣的方式可以輕松達(dá)到目的。但是對于一定變化范圍內(nèi)的索引這樣就不行了。下面的方法是自己想出來的折衷辦法，比較笨，高手清噴。

（2）通過一個(gè)輔助的數(shù)組來解決一定范圍內(nèi)值索引的查找

我們建一個(gè)標(biāo)識元素索引的數(shù)組，然后通過它來顯示符合條件的元素對應(yīng)的索引。還是剛才的數(shù)組，假如我現(xiàn)在要返回元素值在3到100之間值的索引。我可以生成一個(gè)和arr相同大小的數(shù)組，然后首先通過一次篩選找到大于3的元素對應(yīng)索引的數(shù)組，繼而在其中再次篩選，最終得到想要的結(jié)果。代碼如下：

b=np.arange(len(arr))#生成和arr相同長度的數(shù)組 

c=b[arr>3]#c存放的就是arr中大于3的元素對應(yīng)的索引 
#最后通過遍歷c數(shù)組，選擇3到100之間的值打印出來 
for i in range(len(c)): 
 if arr[c[i]]<100: 
  print c[i], 

下面看一下執(zhí)行的效果：

可以看到程序?qū)⑺薪橛?和100之間的元素對應(yīng)的索引值打了出來，如果想同時(shí)獲得索引值和相應(yīng)的元素，只需將上面的“print c[i]”, 替換為 “print c[i],arr[c[i]]”即可。

當(dāng)然這種方法也適用于選擇某個(gè)特定值對應(yīng)的索引，例如我想找到所有3對應(yīng)的位置，可以用print b[arr==3]即可打印出所有值為3的元素所對應(yīng)的索引。其實(shí)不論怎么做，都是用到了數(shù)組進(jìn)行關(guān)系運(yùn)算后生成布爾數(shù)組，然后讓數(shù)組中為True的地方進(jìn)行顯示。

當(dāng)然還可以通過兩次進(jìn)行篩選，分別篩選出大于3的元素和小于100的元素對應(yīng)的索引數(shù)組，然后兩個(gè)數(shù)組做交集處理，在numpy中有一個(gè)intersect1d函數(shù)可以進(jìn)行這樣的運(yùn)算，但是仍然麻煩。目前只能想到這些辦法，不知道有哪位高手有更好的方法，歡迎大家一起交流一下。

以上這篇numpy中實(shí)現(xiàn)ndarray數(shù)組返回符合特定條件的索引方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評論