我的数据框:

 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

任何建议将不胜感激。

最佳答案

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

关于python - 按功能选择由 group 组成的组,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/52661673/

10-16 17:13