Python Pandas 獲取列匹配特定值的行的索引問(wèn)題
給定一個(gè)帶有列"BoolCol"的DataFrame,如何找到滿足條件"BoolCol" == True的DataFrame的索引
目前有迭代的方式來(lái)做到這一點(diǎn):
for i in range(100,3000):
if df.iloc[i]['BoolCol']== True:
print i,df.iloc[i]['BoolCol']
這雖然可行,但不是標(biāo)準(zhǔn)的 Pandas 方式。經(jīng)過(guò)一番研究,我目前正在使用這個(gè)代碼:
df[df['BoolCol'] == True].index.tolist()
這個(gè)給了我一個(gè)索引列表,但跟我想要的不匹配,當(dāng)檢查:
df.iloc[i]['BoolCol']
其結(jié)果實(shí)際上是False!
如何使用正確的 Pandas 方式做到這一點(diǎn)?
最佳解決方法
df.iloc[i]返回df的第i行。 i不引用索引標(biāo)簽,i是從0開(kāi)始的索引。
相反,屬性index返回實(shí)際的索引標(biāo)簽,而不是數(shù)字row-indices:
df.index[df['BoolCol'] == True].tolist()
或者等同地,
df.index[df['BoolCol']].tolist()
通過(guò)使用帶有"unusual"索引的DataFrame,可以非常清楚地看到差異:
df = pd.DataFrame({'BoolCol': [True, False, False, True, True]},
index=[10,20,30,40,50])
In [53]: df
Out[53]:
BoolCol
10 True
20 False
30 False
40 True
50 True
[5 rows x 1 columns]
In [54]: df.index[df['BoolCol']].tolist()
Out[54]: [10, 40, 50]
如果你想使用索引,
In [56]: idx = df.index[df['BoolCol']] In [57]: idx Out[57]: Int64Index([10, 40, 50], dtype='int64')
那么您可以使用loc而不是iloc選擇行:
In [58]: df.loc[idx] Out[58]: BoolCol 10 True 40 True 50 True [3 rows x 1 columns]
請(qǐng)注意,loc也可以接受布爾數(shù)組:
In [55]: df.loc[df['BoolCol']] Out[55]: BoolCol 10 True 40 True 50 True [3 rows x 1 columns]
如果您有一個(gè)布爾數(shù)組mask,并且需要序數(shù)索引值,則可以使用np.flatnonzero來(lái)計(jì)算它們:
In [110]: np.flatnonzero(df['BoolCol']) Out[112]: array([0, 3, 4])
使用df.iloc按順序索引選擇行:
In [113]: df.iloc[np.flatnonzero(df['BoolCol'])] Out[113]: BoolCol 10 True 40 True 50 True python pandas

參考文獻(xiàn)
Python Pandas: Get index of rows which column matches certain value
總結(jié)
以上所述是小編給大家介紹的Python Pandas 獲取列匹配特定值的行的索引問(wèn)題,希望對(duì)大家有所幫助,如果大家有任何疑問(wèn)請(qǐng)給我留言,小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持!
如果你覺(jué)得本文對(duì)你有幫助,歡迎轉(zhuǎn)載,煩請(qǐng)注明出處,謝謝!
相關(guān)文章
Pytorch微調(diào)BERT實(shí)現(xiàn)命名實(shí)體識(shí)別
命名實(shí)體識(shí)別(NER)是自然語(yǔ)言處理(NLP)中的一項(xiàng)關(guān)鍵任務(wù),它涉及識(shí)別和分類文本中的關(guān)鍵實(shí)體,BERT是一種強(qiáng)大的語(yǔ)言表示模型,在各種 NLP 任務(wù)中顯著提高了性能,包括 NER,在本文中,我們將展示如何使用 PyTorch 對(duì)預(yù)訓(xùn)練的 BERT 模型進(jìn)行微調(diào),以用于 NER 任務(wù)2025-03-03
phpsir 開(kāi)發(fā) 一個(gè)檢測(cè)百度關(guān)鍵字網(wǎng)站排名的python 程序
一個(gè)檢測(cè)百度關(guān)鍵字網(wǎng)站排名的python 程序 phpsir 開(kāi)發(fā)2009-09-09
Python實(shí)現(xiàn)AI自動(dòng)玩俄羅斯方塊游戲
提到《俄羅斯方塊》,那真是幾乎無(wú)人不知無(wú)人不曉。其歷史之悠久,可玩性之持久,能手輕輕一揮,吊打一大波游戲。本文將利用Python實(shí)現(xiàn)俄羅斯方塊進(jìn)階版—AI自動(dòng)玩俄羅斯方塊,感興趣的可以學(xué)習(xí)一下2022-03-03
Pythonic版二分查找實(shí)現(xiàn)過(guò)程原理解析
這篇文章主要介紹了Pythonic版二分查找實(shí)現(xiàn)過(guò)程原理解析,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-08-08
詳解OpenCV中簡(jiǎn)單的鼠標(biāo)事件處理
談及鼠標(biāo)事件,就是在觸發(fā)鼠標(biāo)按鈕后程序所做出相應(yīng)的反應(yīng),但是不影響程序的整個(gè)線程。本文將主要介紹OpenCV中的簡(jiǎn)單鼠標(biāo)事件處理,感興趣的可以學(xué)習(xí)一下2022-01-01
python異步編程 使用yield from過(guò)程解析
這篇文章主要介紹了python異步編程 使用yield from過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09
Python使用conda如何安裝requirement.txt的擴(kuò)展包
這篇文章主要介紹了Python使用conda如何安裝requirement.txt的擴(kuò)展包問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-02-02

