Python Pandas 獲取列匹配特定值的行的索引問題
給定一個帶有列"BoolCol"的DataFrame，如何找到滿足條件"BoolCol" == True的DataFrame的索引
目前有迭代的方式來做到這一點：
for i in range(100,3000):
if df.iloc[i]['BoolCol']== True:
print i,df.iloc[i]['BoolCol']
這雖然可行，但不是標準的 Pandas 方式。經(jīng)過一番研究，我目前正在使用這個代碼：
df[df['BoolCol'] == True].index.tolist()
這個給了我一個索引列表，但跟我想要的不匹配，當檢查：
df.iloc[i]['BoolCol']
其結(jié)果實際上是False！
如何使用正確的 Pandas 方式做到這一點？
最佳解決方法
df.iloc[i]返回df的第i行。 i不引用索引標簽，i是從0開始的索引。
相反，屬性index返回實際的索引標簽，而不是數(shù)字row-indices：
df.index[df['BoolCol'] == True].tolist()
或者等同地，
df.index[df['BoolCol']].tolist()
通過使用帶有"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]
請注意，loc也可以接受布爾數(shù)組：
In [55]: df.loc[df['BoolCol']] Out[55]: BoolCol 10 True 40 True 50 True [3 rows x 1 columns]
如果您有一個布爾數(shù)組mask，并且需要序數(shù)索引值，則可以使用np.flatnonzero來計算它們：
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
參考文獻
Python Pandas: Get index of rows which column matches certain value
總結(jié)
以上所述是小編給大家介紹的Python Pandas 獲取列匹配特定值的行的索引問題,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！
相關(guān)文章
Pytorch微調(diào)BERT實現(xiàn)命名實體識別
命名實體識別（NER）是自然語言處理（NLP）中的一項關(guān)鍵任務(wù),它涉及識別和分類文本中的關(guān)鍵實體,BERT是一種強大的語言表示模型,在各種 NLP 任務(wù)中顯著提高了性能,包括 NER,在本文中,我們將展示如何使用 PyTorch 對預(yù)訓練的 BERT 模型進行微調(diào),以用于 NER 任務(wù)2025-03-03
這篇文章主要介紹了使用python實現(xiàn)三維圖可視化，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
phpsir 開發(fā) 一個檢測百度關(guān)鍵字網(wǎng)站排名的python 程序
一個檢測百度關(guān)鍵字網(wǎng)站排名的python 程序 phpsir 開發(fā)2009-09-09
在本篇文章里小編給大家整理的是關(guān)于python中wheel的用法整理內(nèi)容，需要的朋友們可以跟著學習下。2020-06-06
提到《俄羅斯方塊》，那真是幾乎無人不知無人不曉。其歷史之悠久，可玩性之持久，能手輕輕一揮，吊打一大波游戲。本文將利用Python實現(xiàn)俄羅斯方塊進階版—AI自動玩俄羅斯方塊，感興趣的可以學習一下2022-03-03
這篇文章主要介紹了Pythonic版二分查找實現(xiàn)過程原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-08-08
談及鼠標事件，就是在觸發(fā)鼠標按鈕后程序所做出相應(yīng)的反應(yīng)，但是不影響程序的整個線程。本文將主要介紹OpenCV中的簡單鼠標事件處理，感興趣的可以學習一下2022-01-01
這篇文章主要介紹了python異步編程 使用yield from過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-09-09
Python使用conda如何安裝requirement.txt的擴展包
這篇文章主要介紹了Python使用conda如何安裝requirement.txt的擴展包問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-02-02
這篇文章主要介紹了rabbitmq(中間消息代理)在python中的使用詳解，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-12-12