1. 程式人生 > 程式設計 >Python 獲取numpy.array索引值的例項

Python 獲取numpy.array索引值的例項

舉個例子:

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索引值的例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。