通过groupby功能从中选择组 [英] Selecting groups fromed by groupby function

问题描述

我的数据框:

 df1

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

我通过groupby函数形成了每个小组.

I have formed each group bygroupby function.

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

I need to extract the data by using group number.

我想要的输出.

在:获取组0 退出:

   ordercode      quantity
       A             1
       B             3

或

 group     ordercode      quantity
 0            A             1
              B             3

任何建议将不胜感激.

推荐答案

使用 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)

这篇关于通过groupby功能从中选择组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！