在pandas多重索引multiIndex中選定指定索引的行方法
在multiIndex中選定指定索引的行
我們在用pandas類似groupby來使用多重index時,有時想要對多個level中的某個index對應(yīng)的行進(jìn)行操作,就需要在dataframe中找到該index對應(yīng)的行,在單層index中我們可以方便的使用df.loc[index]來選擇,在多重Index中我們可以利用的類似的思路,然而其中也有一些小坑,記錄如下。
1 index為有序的
1.1 創(chuàng)建測試數(shù)據(jù)
首先創(chuàng)建一個dataframe數(shù)據(jù)
df = pd.DataFrame({'class':['A','A','A','B','B','B','C','C'], 'id':['a','b','c','a','b','c','a','b'], 'value':[1,2,3,4,5,6,7,8]})
df中內(nèi)容如下圖:
1.2 設(shè)置multiIndex
通過set_index設(shè)為多重索引
df = df.set_index(['class','id'])
設(shè)置索引后效果:
1.3 切片篩選index
這里同樣使用loc定位
df.loc[('A',slice(None)),:]
各參數(shù)的解釋如下:
loc[(a,b),c]中第一個參數(shù)元組為索引內(nèi)容,a為level0索引對應(yīng)的內(nèi)容,b為level1索引對應(yīng)的內(nèi)容
因為df是一個dataframe,所以要用c來指定列
這里‘A',指選擇class中的A類
slice(None), 是Python中的切片操作,這里用來選擇任意的id,要注意!不能使用‘:'來指定任意index
‘:',用來指定dataframe任意的列
執(zhí)行后的結(jié)果如下:
同樣,如果想只保留id中的'a',則可以使用:
df.loc[(slice(None),'a'),:]
2 index無序
前面的例子對應(yīng)的index列為數(shù)字或字母,是有序的,接下來我們看看index列為中文的情況。
2.1 創(chuàng)建無序測試數(shù)據(jù)
df2 = pd.DataFrame({'課程':['語文','語文','數(shù)學(xué)','數(shù)學(xué)'],'得分':['最高','最低','最高','最低'],'分值':[90,50,100,60]}) df2 = df2.set_index(['課程','得分'])
2.2 嘗試切片選擇index
df2.loc[('語文',slice(None)),:]
我們進(jìn)行同樣的操作,這時會發(fā)現(xiàn)提示出錯:
UnsortedIndexError: 'MultiIndex Slicing requires the index to be fully lexsorted tuple len (2), lexsort depth (0)'
這是因為此時的index無法進(jìn)行排序,在pandas文檔中提到:Furthermore if you try to index something that is not fully lexsorted, this can raise:
我們可以通過 df2.index.is_lexsorted()來檢查index是否有序,
In[1]: df2.index.is_lexsorted() out[1]: False
接下來,我們嘗試對Index進(jìn)行排序。(排序時要在level里指定index名)
2.3 對index排序后切片選擇index
df2 = df2.sort_index(level='課程')
df2.loc[('語文',slice(None)),:]
得到了我們想要的結(jié)果。
參考文獻(xiàn):pandas-docs-MultiIndex / Advanced Indexing
以上這篇在pandas多重索引multiIndex中選定指定索引的行方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python基于queue和threading實現(xiàn)多線程下載實例
這篇文章主要介紹了python基于queue和threading實現(xiàn)多線程下載實例,是比較實用的技巧,需要的朋友可以參考下2014-10-10django框架使用views.py的函數(shù)對表進(jìn)行增刪改查內(nèi)容操作詳解【models.py中表的創(chuàng)建、views.py中
這篇文章主要介紹了django框架使用views.py的函數(shù)對表進(jìn)行增刪改查內(nèi)容操作,結(jié)合實例形式詳細(xì)分析了models.py中表的創(chuàng)建、views.py中函數(shù)的使用,基于對象的跨表查詢等相關(guān)操作技巧與使用注意事項,需要的朋友可以參考下2019-12-12python3 小數(shù)位的四舍五入(用兩種方法解決round 遇5不進(jìn))
這篇文章主要介紹了python3 小數(shù)位的四舍五入(用兩種方法解決round 遇5不進(jìn)),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-04-04Python爬蟲實戰(zhàn)演練之采集糗事百科段子數(shù)據(jù)
讀萬卷書不如行萬里路,只學(xué)書上的理論是遠(yuǎn)遠(yuǎn)不夠的,只有在實戰(zhàn)中才能獲得能力的提升,本篇文章手把手帶你用Python采集糗事百科段子的數(shù)據(jù),大家可以在過程中查缺補漏,提升水平2021-10-10Python創(chuàng)建多線程的兩種常用方法總結(jié)
這篇文章主要為大家詳細(xì)介紹了Python中創(chuàng)建多線程的兩種常用方法,文中的示例代碼簡潔易懂,對我們掌握Python有一定的幫助,需要的可以收藏一下2023-05-05Python爬蟲實戰(zhàn)案例之爬取喜馬拉雅音頻數(shù)據(jù)詳解
這篇文章主要介紹了Python爬蟲實戰(zhàn)案例之取喜馬拉雅音頻數(shù)據(jù)詳解,本文通過實例代碼給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12Python+streamlit實現(xiàn)輕松創(chuàng)建人事系統(tǒng)
streamlit 是 基于 Python 的一個非常強大的 web 構(gòu)建系統(tǒng),通過該類庫,我們可以實現(xiàn)不需要編寫一行前端代碼而構(gòu)建一個完整的 Web 應(yīng)用。下面我們就來編寫一個簡單的人事系統(tǒng)吧2023-02-02