你试过了吗
df.groupby('id').head(2)Ouput生成:
>>> df.groupby('id').head(2) id valueid 1 0 1 1 1 1 2 2 3 2 1 4 2 23 7 3 14 8 4 1(请记住,根据数据,你可能需要先进行订购/排序)
编辑:如发问者所述,用于df.groupby(‘id’).head(2).reset_index(drop=True)去除多义词并展平结果。
>>> df.groupby('id').head(2).reset_index(drop=True) id value0 1 11 1 22 2 13 2 24 3 15 4 1


