pandas：如何根据列分组添加类似索引的列？

Question

我正在尝试将col3添加到现有的pandas数据帧（col1，col2），其中col3就像col1中每个组的索引（从1开始）。

        col1    col2    col3  
0       AAAA    fooW       1
1       AAAA    fooX       2
2       AAAA    fooY       3
3       AAAA    fooZ       4
4       BBBB    fooW       1
5       BBBB    fooX       2
6       CCCC    barV       1
7       CCCC    barW       2
8       CCCC    barX       3
9       CCCC    barY       4
10      CCCC    barZ       5

Answer 1

你首先要分组＆＃39; col1＆＃39;然后应用一些函数来获取组内行的索引。 cumcount是能够胜任这项工作的人。

df['col3'] = df.groupby('col1').cumcount()