Python中的R的seq_len等价物

Question

我正在尝试在我的数据帧在

例如，在这个数据帧中

Col1
A
A
A
B
B
C
D
D
D

我想输出如下

^{pr2}$

在R中，我可以简单地使用数据表df[, idx:=seq_len(.N), by=Col1]。我在Python中找不到等价物。到目前为止，我知道我可以在numpy的包中使用linspace或{}函数，但我不太确定如何按组进行。在

提前谢谢你。在

Answer 1

使用cumcount

In [289]: df['idx'] = df.groupby('Col1').cumcount().add(1)

In [290]: df
Out[290]:
  Col1  idx
0    A    1
1    A    2
2    A    3
3    B    1
4    B    2
5    C    1
6    D    1
7    D    2
8    D    3