pandas - 标记每个 groupby 中的第一个元素

Question

我有一个如下所示的数据框

df = pd.DataFrame({'group':[1,1,2,2,2],'time':[1,2,3,4,5],'C':[6,7,8,9,10]})

    group   time    C
0   1       1       6
1   1       2       7
2   2       3       8
3   2       4       9
4   2       5       10

我希望将每个组中的第一个元素(根据时间)标记为 True，即:

    group   time    C   first_in_group
0   1       1       6   True
1   1       2       7   False
2   2       3       8   True
3   2       4       9   False
4   2       5       10  False

我尝试了 groupby 的几种组合，first 但没有达到我想要的效果。

在 Pandas 中有一种优雅的方法吗？

Answer 1

使用duplicated :

df['first_in_group'] = ~df.group.duplicated()

输出:

   group  time   C  first_in_group
0      1     1   6            True
1      1     2   7           False
2      2     3   8            True
3      2     4   9           False
4      2     5  10           False

注意:首先进行排序(如果需要)。

df = df.sort_values(['group', 'time'])