將序列號添加到每組熊貓python

Question

我有一個簡單的問題，但無法解決。

我有一個簡單的表，例如：

group1 
a
a
a
b
b
b
c
c

我可以使用以下方法向該列添加計數：

df['count'] = range(1, len(df) + 1)

我試圖使用groupby函數更改此設置，但是無法停止計數並按組重新啟動，例如：

group1 count
a 1
a 2
a 3
b 1
b 2
b 3
c 1
c 2

Answer 1

您可以使用cumcount 。 如果要從1開始，可以添加它：

In [16]: df['count'] = df.groupby('group1').cumcount()+1

In [17]: df
Out[17]:
  group1  count
0      a      1
1      a      2
2      a      3
3      b      1
4      b      2
5      b      3
6      c      1
7      c      2