如何修改 groupby.cumcount() 来获取 Pandas 中的计数和出现次数

Question

我正在尝试找到发生的情况，并在重复时添加发生。我尝试了groupby.cumcount，但是当有一个新值时，它不会增加。如果描述令人困惑，我需要如下提到的输出。

	COL1	COL2
0	1200	1
1	1300	2
2	1400	3
3	1400	3
4	1400	3
5	1300	2
6	1500	4
7	1500	4

Answer 1

您正在寻找 。它的每个组从0到groups-1。您可以将1添加到结果系列中，以便编号以1

df['col2'] = df.groupby('col1').ngroup().add(1)

输出启动：

>>> df

   col1
0  1200
1  1300
2  1400
3  1400
4  1400
5  1300
6  1500
7  1500

>>> df.groupby('col1').ngroup()

0    0
1    1
2    2
3    2
4    2
5    1
6    3
7    3
dtype: int64

>>> df.groupby('col1').ngroup().add(1)

0    1
1    2
2    3
3    3
4    3
5    2
6    4
7    4
dtype: int64