根据数据帧中的条件删除组 - pandas groupby

Amir 发表于 Dev

阿米尔

这是我的数据框：

df = pd.DataFrame({'sym': list('aaaaaabb'), 'order': [0, 0, 1, 1, 0, 1, 0, 1], 'key': [2, 2, 2, 2, 3, 3, 4, 4],
                   'vol': [1000, 1000, 500, 500, 100, 100, 200, 200]})

我向其中添加另一列：

df['vol_cumsum'] = df.groupby(['sym', 'key', 'order']).vol.cumsum()

让我们这样定义问题（而不是文字）。检查这个：

df.groupby(['sym', 'key', 'order']).vol_cumsum.last()

现在我想省略vol_cumsum根据上述 groupby 不匹配的组。在这种情况下，我想从我的df. 我想要的df看起来像这样：

4    3      0   a   100         100
5    3      1   a   100         100
6    4      0   b   200         200
7    4      1   b   200         200

耶斯列

使用GroupBy.transformwith GroupBy.lastforSeries与 original 大小相同DaatFrame，然后通过DataFrame.assignwith创建 nw 列GroupBy.all：

df['vol_cumsum'] = df.groupby(['sym', 'key', 'order']).vol.cumsum()
s = df.groupby(['sym', 'key', 'order']).vol_cumsum.transform('last')
mask = df.assign(new=df['vol_cumsum'].eq(s)).groupby(['sym', 'key', 'order'])['new'].transform('all')

df = df[mask]
print (df)
  sym  order  key  vol  vol_cumsum
4   a      0    3  100         100
5   a      1    3  100         100
6   b      0    4  200         200
7   b      1    4  200         200

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。