numpy高級(jí)索引

numpy比一般的python序列提供更多的索引方式。除了之前看到的用整數(shù)和切片的索引外,數(shù)組可以用整數(shù)數(shù)組索引、布爾索引及花式索引。整數(shù)數(shù)組索引

實(shí)例1:

獲取數(shù)組中(0,0),(1,1)和(2,0)位置處的元素

import numpyas np

a=np.array([[1,2,3],[3,4,5],[5,6,7]])

y=a[[0,1,2],[0,1,0]]

print(y)


獲取多維數(shù)組的四個(gè)角的元素。

實(shí)例2:

import numpyas np

a=np.array([[1,2,3],[4,5,6],[7,8,9],[0,1,2]])

b=np.array([[0,0],[3,3]])

d=np.array([[0,2],[0,2]])

y=a[b,d]

print(y)


借助切片:或...與索引數(shù)組組合。

實(shí)例3:

import numpyas np

a=np.array([[1,2,3],[4,5,6],[7,8,9]])

b=a[1:3,1:3]

print(b)

print('\n')

c=a[1:3,(1,2)]

print(c)

print('\n')

d=a[...,1:]

print(d)

print('\n')


布爾索引

我們可以通過一個(gè)布爾數(shù)組來(lái)索引目標(biāo)數(shù)組。

布爾索引通過布爾運(yùn)算(如:比較運(yùn)算符)來(lái)獲取符合指定條件的元素的數(shù)組。

實(shí)例4:大于5的元素

import numpyas np

a=np.array([[1,2,3],[4,5,6],[7,8,9]])

b=a[a>5]

print(b)


實(shí)例5:使用~(取補(bǔ)運(yùn)算符)過濾NaN

import numpyas np

a=np.array([np.nan,1,2,np.nan,3,4,5])

print(a[~np.isnan(a)])


實(shí)例6:過濾非復(fù)數(shù)元素

import numpyas np

a=np.array([1,2+6j,5,3.5+5j])

print(a[np.iscomplex(a)])


花式索引

花式索引指的是利用整數(shù)數(shù)組進(jìn)行索引

花式索引根據(jù)索引數(shù)組的值作為目標(biāo)數(shù)組的某個(gè)軸的下標(biāo)來(lái)取值。對(duì)于使用一維整型數(shù)組作為索引,如果目標(biāo)是一維數(shù)組,name索引的結(jié)果就是對(duì)應(yīng)衛(wèi)位置的元素;如果目標(biāo)是二維數(shù)組,那么就是對(duì)應(yīng)下標(biāo)的行。

花式索引跟切片不一樣,它總是將數(shù)據(jù)復(fù)制到新數(shù)組中。

實(shí)例7:傳入順序索引數(shù)組

import numpyas np

x=np.arange(32).reshape((8,4))

print(x)

print(x[[4,2,1,7]]) //取的對(duì)應(yīng)的行

實(shí)例8:傳入倒序索引數(shù)組

import numpyas np

x=np.arange(32).reshape((8,4))

print(x)

print('\n')

print(x[[-4,-2,-1,-7]])


實(shí)例9:傳入多個(gè)索引數(shù)組(要使用np.ix_)

import numpyas np

x=np.arange(32).reshape((8,4))

print(x)

print('\n')

print(x[np.ix_([1,5,7,2],[0,3,1,2])])


最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容