Trancher pandas géants multiindice dataframe en utilisant la liste des valeurs de l'indice

Question

J'ai un dataframe de multiindice qui ressemble à

uid texte tid

abc x t1

DCB y t2

uid et tid sont les index. J'ai une liste de uid s, et je veux obtenir les rangées correspondant aux uids dans cette liste, mais en gardant les valeurs d'index de deuxième niveau (tid). Je veux le faire sans exécuter de boucle explicite. Est-ce possible?

Answer 1

données:

L = ['abc', 'bcd'] 

print (df) 
     text 
uid tid  
abc x  t1 
abc1 x  t1 
bcd y  t2 

1. slicers

idx = pd.IndexSlice 
df1 = df.loc[idx[L,:],:] 

---

2. boolean indexing + masque avec get_level_values + isin:

df1 = df[df.index.get_level_values(0).isin(L)] 

---

3. query, docs:

df1 = df.query('@L in uid') 

print (df1) 
     text 
uid tid  
abc x  t1 
bcd y  t2