Python 獲取numpy.array索引值的例項
阿新 • • 發佈:2020-01-09
舉個例子:
q=[0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
我想獲取其中值等於7的那個值的下標,以便於用於其他計算。
如果使用np.where,如:
q=np.arange(0,16,1) g=np.where(q==7) print q print g
執行結果是:
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15] (array([7]),)
顯然(array([7]),)中的數字7我是沒法提取出來做運算的,這是一個tuple。
處理方法是:
q=np.arange(0,1) g=np.argwhere(q==7)//注意這裡,返回的是下標而不是索引值 print q print g print int(g)//轉化為可運算的數字
執行結果為:
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15] [[7]] 7
以上這篇Python 獲取numpy.array索引值的例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。