python - 通过 groupby 函数选择组

Question

我的数据框:

 df1

group     ordercode      quantity
 0            A             1
              B             3
 1            C             1
              E             2
              D             1

我已经通过groupby 函数形成了每个组。

我需要使用组号提取数据。

我想要的输出。

在:获取第0组出:

   ordercode      quantity
       A             1
       B             3

或

 group     ordercode      quantity
 0            A             1
              B             3

如有任何建议，我们将不胜感激。

Answer 1

使用DataFrame.xs , 也可以使用参数 drop_level=False:

#if need remove original level

df1 = df.xs(0)
print (df1)
           quantity
ordercode          
A                 1
B                 3

#if avoid remove original level
df1 = df.xs(0, drop_level=False)
print (df1)
                 quantity
group ordercode          
0     A                 1
      B                 3

编辑:

dfs = [df1, df2, df3]
dfs = [x[x['group'] == 0] for x in dfs]
print (dfs)