如何按组使熊猫列连续？

Question

我的数据框看起来像

ID     Number
 1       2
 1       1
 1       4
 2       1
 2       3
 2       4
 2       5

然后，我想创建另一个名为“连续数字”的列，看起来像

ID     Number   Consecutive Number
 1       2              2
 1       1              1
 1       4              3
 2       1              1
 2       3              2
 2       4              3
 2       5              4

有没有简单的方法来实现这一点？

Answer 1

试试rank

df['new'] = df.groupby('ID').Number.rank()

Out[50]: 
0    2.0
1    1.0
2    3.0
3    1.0
4    2.0
5    3.0
6    4.0
Name: Number, dtype: float64