我有一個(gè)帶有 3 個(gè)級(jí)別的 MultiIndex 的 Pandas DataFrame。假設(shè)我有以下數(shù)據(jù)df = pd.DataFrame({'ColB': {('A1', 'B1', 1): 'cb1', ('A1', 'B1', 2): 'cb2', ('A1', 'B2', 1): 'cb3', ('A1', 'B2', 2): 'cb4', ('A2', 'B1', 1): 'cb5', ('A2', 'B1', 2): 'cb6', ('A2', 'B2', 1): 'cb7', ('A2', 'B2', 2): 'cb8'}, 'colA': {('A1', 'B1', 1): 'ca1', ('A1', 'B1', 2): 'ca2', ('A1', 'B2', 1): 'ca3', ('A1', 'B2', 2): 'ca4', ('A2', 'B1', 1): 'ca5', ('A2', 'B1', 2): 'ca6', ('A2', 'B2', 1): 'ca7', ('A2', 'B2', 2): 'ca8'}}) ColB colAA1 B1 1 cb1 ca1 2 cb2 ca2 B2 1 cb3 ca3 2 cb4 ca4A2 B1 1 cb5 ca5 2 cb6 ca6 B2 1 cb7 ca7 2 cb8 ca8現(xiàn)在，我有一個(gè) MultiIndex 對(duì)象，其中包含前兩個(gè)級(jí)別的索引，例如MultiIndex([('A1', 'B2'), ('A2', 'B1')], )我想使用該 MultiIndex 來選擇與該 MultiIndex 對(duì)應(yīng)的所有行，包括第 3 級(jí)的所有索引，例如， ColB colAA1 B2 1 cb3 ca3 2 cb4 ca4A2 B1 1 cb5 ca5 2 cb6 ca6我怎樣才能做到這一點(diǎn)？我一直在尋找?guī)讉€(gè)小時(shí)的答案，但我仍然毫無頭緒。謝謝。
3 回答
湖上湖
TA貢獻(xiàn)2003條經(jīng)驗(yàn) 獲得超2個(gè)贊
使用Index.isin刪除第 3 級(jí)并MultiIndex.droplevel過濾boolean indexing：
df = df[df.index.droplevel(2).isin(mux)]
print (df)
? ? ? ? ColB colA
A1 B2 1? cb3? ca3
? ? ? 2? cb4? ca4
A2 B1 1? cb5? ca5
? ? ? 2? cb6? ca6
它適用于任何index：
mux = pd.MultiIndex.from_tuples([('A1', 'B1'),('A2', 'B2')])
df = df[df.index.droplevel(2).isin(mux)]
print (df)
? ? ? ? ColB colA
A1 B1 1? cb1? ca1
? ? ? 2? cb2? ca2
A2 B2 1? cb7? ca7
? ? ? 2? cb8? ca8
拉風(fēng)的咖菲貓
TA貢獻(xiàn)1995條經(jīng)驗(yàn) 獲得超2個(gè)贊
咱們?cè)囋嚢?code>Advanced indexing with hierarchical index
df.loc[('A1', 'B2'):('A2','B1')]
Out[56]:?
? ? ? ? ColB colA
A1 B2 1? cb3? ca3
? ? ? 2? cb4? ca4
A2 B1 1? cb5? ca5
? ? ? 2? cb6? ca6
添加回答
舉報(bào)
0/150
提交
取消