python - 如何在GroupBy对象的每个组中删除某个索引处的行？

Question

我试图在 GroupBy 对象内的每个组中的某个索引处删除一行。
我能够管理的最好的是:

import pandas as pd
    
x_train = x_train.groupby('ID')
x_train.apply(lambda x: x.drop([0], axis=0))

但是，这不起作用。我花了一整天的时间都没有解决方案，所以转向堆栈。
编辑:还需要任何索引值的解决方案

Answer 1

你可以用 cumcount 来做

idx= x_train.groupby('ID').cumcount()
x_train = x_train[idx!=0]